hdu5242 Game (贪心+dfs序)

dfs序后用线段树来记每个节点的前缀和

每次找一个前缀和最大的节点，然后把它到根的路径上的每个之前没被走过的点 对应的dfs序的区间 减掉那个点的权值

每个点最多被减一次，复杂度是$O(nlogn)$的

 #include<bits/stdc++.h>
 #define pa pair<int,int>
 #define CLR(a,x) memset(a,x,sizeof(a))
 using namespace std;
 typedef long long ll;
 const int maxn=1e5+;

 inline ll rd(){
     ll x=;char c=getchar();int neg=;
     while(c<''||c>''){if(c=='-') neg=-;c=getchar();}
     while(c>=''&&c<='') x=x*+c-'',c=getchar();
     return x*neg;
 }

 int eg[maxn][],egh[maxn],ect,root,fa[maxn];
 int N,K,ch[maxn][],pct,dfn[maxn][],tot,id[maxn];
 ll v[maxn],dis[maxn],ma[maxn],mi[maxn];
 ll laz[maxn];
 bool flag[maxn];

 inline void adeg(int a,int b){
     eg[++ect][]=b;eg[ect][]=egh[a];egh[a]=ect;
 }

 void dfs(int x){
     dfn[x][]=++tot,id[tot]=x;

     for(int i=egh[x];i;i=eg[i][]){
         int b=eg[i][];if(b==fa[x]) continue;
         fa[b]=x;dis[b]=dis[x]+v[b];
         dfs(b);
     }
     dfn[x][]=tot;
 }

 inline void update(int p){
     if(!ch[p][]) return;
     if(ma[ch[p][]]>ma[ch[p][]]) ma[p]=ma[ch[p][]],mi[p]=mi[ch[p][]];
     else ma[p]=ma[ch[p][]],mi[p]=mi[ch[p][]];
 }
 inline void pushdown(int p){
     if(!laz[p]||!ch[p][]) return;
     ma[ch[p][]]+=laz[p],ma[ch[p][]]+=laz[p];
     laz[ch[p][]]+=laz[p],laz[ch[p][]]+=laz[p];
     laz[p]=;
 }

 void build(int &p,int l,int r){
     p=++pct;
     if(l==r) ma[p]=dis[id[l]],mi[p]=id[l];
     else{
         int m=l+r>>;
         build(ch[p][],l,m);build(ch[p][],m+,r);
         update(p);
     }
 }

 void add(int p,int l,int r,int x,int y,ll z){
     pushdown(p);
     if(x<=l&&r<=y){
         laz[p]+=z;ma[p]+=z;
         pushdown(p);
     }else{
         int m=l+r>>;
         if(x<=m) add(ch[p][],l,m,x,y,z);
         if(y>=m+) add(ch[p][],m+,r,x,y,z);
         update(p);
     }
 }

 int main(){
     //freopen("","r",stdin);
     int i,j,k;
     int T=rd();
     for(int tt=;tt<=T;tt++){
         N=rd(),K=rd();
         CLR(ch,);CLR(egh,);CLR(laz,);ect=pct=tot=root=;
         CLR(dis,);CLR(flag,);
         for(i=;i<=N;i++) v[i]=rd();
         for(i=;i<N;i++){
             int a=rd(),b=rd();
             adeg(a,b);
         }
         dis[]=v[];dfs();
         build(root,,N);
         ll ans=;
         for(i=;i<=K;i++){
             ans+=ma[root];int ii=mi[root];
             // printf("!%lld %d\n",ma[root],mi[root]);
             while(ii&&!flag[ii]){
                 add(root,,N,dfn[ii][],dfn[ii][],-v[ii]);
                 // printf("--%d %d %d\n",dfn[ii][0],dfn[ii][1],v[ii]);
                 flag[ii]=;ii=fa[ii];
             }
         }
         printf("Case #%d: %lld\n",tt,ans);
     }
     return ;
 }