dfs序+线段树 BZOJ3252 攻略

3252: 攻略

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 496  Solved: 211
[Submit][Status][Discuss]

Description

题目简述：树版[k取方格数]

 

众所周知，桂木桂马是攻略之神，开启攻略之神模式后，他可以同时攻略k部游戏。

今天他得到了一款新游戏《XX半岛》，这款游戏有n个场景(scene)，某些场景可以通过不同的选择支到达其他场景。所有场景和选择支构成树状结构：开始游戏时在根节点（共通线），叶子节点为结局。每个场景有一个价值，现在桂马开启攻略之神模式，同时攻略k次该游戏，问他观赏到的场景的价值和最大是多少（同一场景观看多次是不能重复得到价值的）

“为什么你还没玩就知道每个场景的价值呢？”

“我已经看到结局了。”

Input

第一行两个正整数n,k

第二行n个正整数，表示每个场景的价值

以下n-1行,每行2个整数a,b，表示a场景有个选择支通向b场景（即a是b的父亲）

保证场景1为根节点

Output

 

输出一个整数表示答案

Sample Input

5 2
4 3 2 1 1
1 2
1 5
2 3
2 4

Sample Output

10

HINT

对于100%的数据，n<=200000，1<=场景价值<=2^31-1

 

 

dfs序+线段树的模板吧……

函数传参不要写错！！！！！！

 

  1 #include<iostream>  2 #include<cstdio>  3 #include<cstring>  4 #include<algorithm>  5 using namespace std;  6 #define inf 1000000000000000ll   7 int n,k,cnt,tot;  8 long long ans;  9 long long val[200010],sum[200010]; 10 int head[200010],getin[200010],getout[200010],oula[200010],fa[200010]; 11 bool check[200010]; 12 struct data{ 13     int next,to; 14 }edge[400010]; 15 struct dt{ 16     long long mx,loc,tag; 17 }segtree[800010];  18 void add(int start,int end){ 19     edge[++cnt].next=head[start]; 20     edge[cnt].to=end; 21     head[start]=cnt; 22     fa[end]=start; 23 } 24 void dfs(int u){ 25     getin[u]=++tot; 26     oula[tot]=u; 27     for(int i=head[u];i;i=edge[i].next){ 28         sum[edge[i].to]=sum[u]+val[edge[i].to]; 29         dfs(edge[i].to); 30     } 31     getout[u]=tot; 32 } 33 void up(int pos){ 34     segtree[pos].mx=0; 35     if(segtree[pos<<1].mx>segtree[pos].mx){ 36         segtree[pos].mx=segtree[pos<<1].mx; 37         segtree[pos].loc=segtree[pos<<1].loc; 38     } 39     if(segtree[pos<<1|1].mx>segtree[pos].mx){ 40         segtree[pos].mx=segtree[pos<<1|1].mx; 41         segtree[pos].loc=segtree[pos<<1|1].loc;        42     } 43 } 44 void build(int pos,int ll,int rr){ 45     if(ll==rr){ 46         segtree[pos].mx=sum[oula[ll]]; 47         segtree[pos].loc=ll; 48         return; 49     } 50     int mid=(ll+rr)>>1; 51     build(pos<<1,ll,mid); 52     build(pos<<1|1,mid+1,rr); 53     up(pos); 54 } 55 void update(int pos,long long dd){//!!!!!!!!!!!!!!!!!!!!!!!! long long 56     segtree[pos].mx-=dd; 57     segtree[pos].tag+=dd; 58 } 59 void down(int pos){ 60     if(!segtree[pos].tag) return; 61     update(pos<<1,segtree[pos].tag); 62     update(pos<<1|1,segtree[pos].tag); 63     segtree[pos].tag=0; 64     return; 65 } 66 void change(int pl,int pr,long long dd,int pos,int ll,int rr){ 67     int mid=(ll+rr)>>1; 68     if(pl>rr||pr<ll) return; 69     if(pl<=ll&&pr>=rr){ 70         update(pos,dd); 71         return; 72     } 73     down(pos); 74     if(pr<=mid) change(pl,pr,dd,pos<<1,ll,mid); 75     else if(pl>mid) change(pl,pr,dd,pos<<1|1,mid+1,rr); 76     else{ 77         change(pl,mid,dd,pos<<1,ll,mid); 78         change(mid+1,pr,dd,pos<<1|1,mid+1,rr); 79     } 80     up(pos); 81 } 82 int main(){ 83     scanf("%d%d",&n,&k); 84     for(int i=1;i<=n;i++) scanf("%lld",&val[i]); 85     int x,y; 86     for(int i=1;i<n;i++){ 87         scanf("%d%d",&x,&y); 88         add(x,y); 89     } 90     sum[1]=val[1]; 91     dfs(1); 92     build(1,1,n); 93     for(int i=1;i<=k;i++){ 94         ans+=segtree[1].mx; 95         for(int j=oula[segtree[1].loc];!check[j]&&j;j=fa[j]){ 96             check[j]=1; 97             change(getin[j],getin[j],inf,1,1,n);//ll的极值？？？  98             if(getin[j]<getout[j]) change(getin[j]+1,getout[j],val[j],1,1,n); 99         }100     }101     printf("%lld",ans);102     return 0;103 }