【DFS序+线段树区间更新区间求最值】HDU 5692 Snacks

http://acm.hdu.edu.cn/showproblem.php?pid=5692

【思路】

• 每更新一个点，子树的所有结点都要更新，所以是区间更新
• 每查询一个点，子树的所有结点都要查询，所以是区间查询最值
• 线段树上结点的值是每个点到根这条链的权值和

【AC】

 #pragma comment(linker, "/STACK:1024000000,1024000000")
 #include<bits/stdc++.h>
 #define lson l,mid,rt<<1
 #define rson mid+1,r,rt<<1|1
 using namespace std;
 typedef long long ll;
 int n,m;
 const int maxn=1e5+;
 const int maxm=*maxn;
 const ll linf=1e18;
 struct edge
 {
     int to;
     int nxt;
 }e[maxm];
 int head[maxn];
 int tot;
 int cid;
 int st[maxn],ed[maxn];
 ll tree[maxn];
 ll a[maxn];
 ll lazy[maxn*];
 ll sum[maxn*];
 int que[maxn];
 void init()
 {
     memset(head,-,sizeof(head));
     tot=;
     cid=;
     memset(tree,,sizeof(tree));
     memset(lazy,,sizeof(lazy));
 }
 void addedge(int u,int v)
 {
     e[tot].to=v;
     e[tot].nxt=head[u];
     head[u]=tot++;
 }
 void dfs(int u,int pa)
 {
     st[u]=++cid;
     que[cid]=u;
     for(int i=head[u];i!=-;i=e[i].nxt)
     {
         int v=e[i].to;
         if(v==pa) continue;
         tree[v]=tree[u]+a[v];
         dfs(v,u);
     }
     ed[u]=cid;
 }
 void pushup(int rt)
 {
     sum[rt]=max(sum[rt<<],sum[rt<<|]);
     return;
 }
 void pushdown(int rt)
 {
     if(lazy[rt])
     {
         sum[rt<<]+=lazy[rt];
         sum[rt<<|]+=lazy[rt];
         lazy[rt<<]+=lazy[rt];
         lazy[rt<<|]+=lazy[rt];
         lazy[rt]=;
     }
 }
 void build(int l,int r,int rt)
 {
     lazy[rt]=;
     if(l==r)
     {
         sum[rt]=tree[que[l]];
         return;
     }
     int mid=(l+r)>>;
     build(lson);
     build(rson);
     pushup(rt);
 }

 void update(int L,int R,ll val,int l,int r,int rt)
 {
     if(L<=l && r<=R)
     {
         lazy[rt]+=val;
         sum[rt]+=val;
         return;
     }
     int mid=(l+r)>>;
     pushdown(rt);
     if(L<=mid)
     {
         update(L,R,val,lson);
     }
     if(mid<R)
     {
         update(L,R,val,rson);
     }
     pushup(rt);
 }
 ll query(int L,int R,int l,int r,int rt)
 {
     if(L<=l && r<=R)
     {
         return sum[rt];
     }
     pushdown(rt);
     int mid=(l+r)>>;
     ll ans=-linf;
     if(L<=mid)
     {
         ans=max(ans,query(L,R,lson));
     }
     if(mid<R)
     {
         ans=max(ans,query(L,R,rson));
     }
     return ans;
 }
 int main()
 {
     int T;
     scanf("%d",&T);
     int cas=;
     while(T--)
     {
         scanf("%d%d",&n,&m);
         init();
         int u,v;
         for(int i=;i<=n-;i++)
         {
             scanf("%d%d",&u,&v);
             u++;v++;
             addedge(u,v);
             addedge(v,u);
         }
         for(int i=;i<=n;i++)
         {
             scanf("%I64d",&a[i]);
         }
         tree[]=a[];
         dfs(,-);
         build(,cid,);
         printf("Case #%d:\n",++cas);
         while(m--)
         {
             int tp;
             scanf("%d",&tp);
             if(tp==)
             {
                 int x;ll val;
                 scanf("%d%I64d",&x,&val);
                 x++;
                 update(st[x],ed[x],val-a[x],,cid,);
                 a[x]=val;
             }
             else
             {
                 int x;scanf("%d",&x);x++;
                 ll ans=query(st[x],ed[x],,cid,);
                 printf("%I64d\n",ans);
             }
         }
     }
     return ;
 } 

【注意】

线段树要开4倍空间，因为这个WA了好久