攻略 bzoj-3252

题目大意:给定一棵n个节点的有根树,点有点权。让你选出至多k个节点,使得他们到根的链的并最大。

注释:$1\le n\le 2\cdot 10^5$,$1\le val_i\le 2^{31}-1$。


想法:这题模拟赛T2,正解可并堆,我用的$dfs$序加线段树。

考虑暴力:显然每次选取当前贡献最大的一定是最优的,一个非常显然的贪心。

我们对每个节点维护一个$dis$表示当前节点选取的贡献。如果我们选取了一个节点$x$那么以$x$为根的子树都会减去$val_x$。

显然一个节点只会被选取一次,所以如果我们每次可以快速找到最大值的位置,我们就可以对以$x$为根的子树进行子树减。

故此我们在线段树上的每个节点都维护一个$son$信息,表示线段树上的最大值是左儿子贡献的还是右儿子贡献的。

对于线段树上的叶子节点再维护一下这个叶子对应原树的哪个节点。

这样的话每次我们从根开始通过那个$son$信息找到最大的贡献节点,然后从当前节点开始向根遍历知道遇到被选取过的节点。每遍历到一个节点都进行一遍子树减即可。

总时间复杂度$O(nlogn)$。

Code:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 200010
#define ls p<<1
#define rs p<<1|1
using namespace std; typedef long long ll;
struct Node
{
ll mx,del; int id,son;
}a[N<<2];
ll dis[N]; int dic[N],cnt,val[N],re[N],f[N],size[N];
int to[N<<1],nxt[N<<1],head[N],tot;
bool vis[N];
inline char nc() {static char *p1,*p2,buf[100000]; return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;}
ll rd() {ll x=0; char c=nc(); while(!isdigit(c)) c=nc(); while(isdigit(c)) x=(x<<3)+(x<<1)+(c^48),c=nc(); return x;}
inline void add(int x,int y) {to[++tot]=y; nxt[tot]=head[x]; head[x]=tot;}
inline void pushup(int p)
{
a[p].mx=max(a[ls].mx,a[rs].mx);
a[p].son=(a[ls].mx>=a[rs].mx?1:2);
}
inline void pushdown(int p)
{
if(!a[p].del) return;
a[ls].mx+=a[p].del; a[ls].del+=a[p].del;
a[rs].mx+=a[p].del; a[rs].del+=a[p].del;
a[p].del=0;
}
void build(int l,int r,int p)
{
if(l==r) {a[p].mx=dis[re[l]],a[p].id=re[l]; return;}
int mid=(l+r)>>1;
build(l,mid,ls); build(mid+1,r,rs);
pushup(p);
}
void update(int x,int y,ll val,int l,int r,int p)
{
if(x<=l&&r<=y)
{
a[p].mx+=val; a[p].del+=val;
return;
}
int mid=(l+r)>>1; pushdown(p);
if(x<=mid) update(x,y,val,l,mid,ls);
if(mid<y) update(x,y,val,mid+1,r,rs);
pushup(p);
}
int query_son(int l,int r,int p)
{
if(l==r) return p;
int mid=(l+r)>>1; pushdown(p);
if(a[p].son==1) return query_son(l,mid,ls);
else return query_son(mid+1,r,rs);
}
void dfs(int pos,int fa)
{
dis[pos]=dis[fa]+val[pos];
f[pos]=fa;
dic[pos]=++cnt; re[cnt]=pos;
size[pos]=1;
for(int i=head[pos];i;i=nxt[i]) if(to[i]!=fa)
{
dfs(to[i],pos);
size[pos]+=size[to[i]];
}
}
int main()
{
ll ans=0;
int n=rd(),k=rd();
for(int i=1;i<=n;i++) val[i]=rd();
for(int i=1;i<n;i++)
{
int x=rd(),y=rd(); add(x,y); add(y,x);
}
dfs(1,0);
build(1,n,1);
while(k--)
{
int p=query_son(1,n,1);
ans+=a[p].mx;
int x=a[p].id;
while(x&&!vis[x])
{
update(dic[x],dic[x]+size[x]-1,-val[x],1,n,1); val[x]=0;
vis[x]=true;
x=f[x];
}
}
cout << ans << endl ;
// fclose(stdin); fclose(stdout);
return 0;
}

小结:从暴力入手是一个非常好的选择。

[bzoj3252]攻略_dfs序_线段树_贪心的更多相关文章

  1. bzoj3252 攻略 dfs序+线段树

    题目传送门 题目大意:给出一棵树,1为根节点,每个节点都有权值,每个叶子节点都是一个游戏的结局,选择k个游戏结局,使得权值总和最大,同一个节点不会被重复计算. 思路:这道题最关键的是要想到一个性质,就 ...

  2. [bzoj3306]树_dfs序_线段树_倍增lca

    树 bzoj-3306 题目大意:给定一颗n个节点的树,支持换根.修改点权.查询子树最小值. 注释:$1\le n,q\le 10^5$. 想法: 如果没有换根操作,就是$dfs$序+线段树维护区间最 ...

  3. [Cometoj#3 D]可爱的菜菜子_线段树_差分_线性基

    可爱的菜菜子 题目链接:https://cometoj.com/contest/38/problem/D?problem_id=1543 数据范围:略. 题解: 首先,如果第一个操作是单点修改,我们就 ...

  4. [LuoguP3064][USACO12DEC]伊斯坦布尔的帮派Gangs of Istanbull(加强版)_线段树_贪心

    伊斯坦布尔的帮派Gangs of Istanbull 题目链接:https://www.luogu.org/problem/P3064 数据范围:略. 题解: 这个题其实分为两问,第一问是$YES$. ...

  5. [bzoj2962]序列操作_线段树_区间卷积

    序列操作 bzoj-2962 题目大意:给定一个n个数的正整数序列,m次操作.支持:1.区间加:2.区间取相反数:3.区间求选c个数的乘积和. 注释:$1\le n,m\le 5\cdot 10^4$ ...

  6. [计蒜客T2238]礼物_线段树_归并排序_概率期望

    礼物 题目大意: 数据范围: 题解: 这题有意思啊($md$卡常 直接做怎么做? 随便上个什么东西,维护一下矩阵乘和插入,比如说常数还算小的$KD-Tree$(反正我是没见人过过 我们漏掉了一个条件, ...

  7. [bzoj5379]Tree_dfs序_线段树_倍增lca

    Tree bzoj-5379 题目大意:给定一棵$n$节点的树.支持:换根.把节点$u$和$v$的$lca$的子树加.询问$u$的子树和. 注释:$1\le n,q\le 3\times 10^5$. ...

  8. 洛谷 P2542 [AHOI2005]航线规划 树链剖分_线段树_时光倒流_离线

    Code: #include <map> #include <cstdio> #include <algorithm> #include <cstring&g ...

  9. BZOJ_3252_攻略_线段树+dfs序

    BZOJ_3252_攻略_线段树+dfs序 Description 题目简述:树版[k取方格数] 众所周知,桂木桂马是攻略之神,开启攻略之神模式后,他可以同时攻略k部游戏.今天他得到了一款新游戏< ...

随机推荐

  1. Vue.js学习笔记--4. 组件的基本使用

    整理自官网教程 -- https://cn.vuejs.org/ 1. 所有Vue组件同时也都是Vue实例,分为全局组件和局部组件,注册方式如下. <div id="app" ...

  2. 轻松搞定Spring+quartz的定时任务

    1.spring 的定时任务写法有两种:一种是继承工作类,一种是普通的Bean,定时写法有两种写法:一种是以时间间隔启动任务SimpleTriggerBean,一种是以时刻启动任务CronTrigge ...

  3. vue2.0版本指令v-if与v-show的区别

    v-if: 判断是否加载,可以减轻服务器的压力,在需要时加载. v-show:调整css dispaly属性,可以使客户端操作更加流畅. v-if示例: <!DOCTYPE html> & ...

  4. vuex使用之state访问状态对象

    引入vuex1.利用npm包管理工具,进行安装 vuex.在控制命令行中输入下边的命令就可以了. npm install vuex --save 需要注意的是这里一定要加上 –save,因为你这个包我 ...

  5. PostgreSQL 数据库错误状态编号解释[附带列表

    PostgreSQL 服务器发出的所有消息都赋予了五个字符的错误代码, 这些代码遵循 SQL 的 "SQLSTATE" 代码的习惯.需要知道发生了什么错误条件的应用通常应该测试错误 ...

  6. 数据结构算法 - ConcurrentHashMap 源码解析

    五个线程同时往 HashMap 中 put 数据会发生什么? ConcurrentHashMap 是怎么保证线程安全的? 在分析 HashMap 源码时还遗留这两个问题,这次我们站在 Java 多线程 ...

  7. jvm中的内存溢出与内存泄露

    内存溢出: 就是我们通常遇到的OutOfMemoryError异常,它俗理解就是内存不够,通常在运行大型程序时发生,当程序所需要的内存远远超出了JVM内存所承受大小,就会报出OutOfMemoryEr ...

  8. python3+beautifulSoup4.6抓取某网站小说(一)爬虫初探

    本次学习重点: 1.使用urllib的request进行网页请求,获取当前url整版网页内容 2.对于多级抓取,先想好抓取思路,再动手 3.BeautifulSoup获取html网页中的指定内容 4. ...

  9. sublime text 快捷键记录

    sublime常用快捷键 Ctrl+D 选词 (反复按快捷键,即可继续向下同时选中下一个相同的文本进行同时编辑) Ctrl+G 跳转到相应的行 Ctrl+J 合并行(已选择需要合并的多行时) Ctrl ...

  10. 题解 洛谷P2147/BZOJ2049【[SDOI2008]洞穴勘测】

    Link-Cut-Tree的模板题啊......(听说还可以用其他的方法做,不管了,直接上LCT) 没有要求维护点权,只需要维护点的连通性即可. 就是朴素的LCT,居然还不要pushup. 感觉有些不 ...