BZOJ1984: 月下“毛景树”

BZOJ 1984: 月下“毛景树”

Time Limit: 20 Sec Memory Limit: 64 MB

Submit: 1583 Solved: 500

[Submit][Status][Discuss]

题解：
1.边权转点权：先建树，可以将边权转换成f和t中深度较大的点，也就是每个结点的权值都是代表与父亲结点连边后边上的权值
（最后修改的时候注意最后一个点的点权不要加，因为那不是所求链上的边）
2.双标记判断：先对覆盖做pushdown，有覆盖标记时先把加法的标记清0

<span style="font-family:Microsoft YaHei;"><span style="font-weight: normal;"><span style="font-size:14px;">#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=400005;
struct Tree {int l,r,cov,plu,mx;}T[N*4];
int n;
int ff[N],tt[N],pp[N];
int fa[N],pos[N],npos[N],son[N],s[N],dep[N],top[N],dian[N];
int cnt,to[N],nxt[N],lj[N],a[N];
void pushdown(int x)
{int lc=x<<1,rc=x<<1|1;if(T[x].cov>=0){T[lc].mx=T[x].cov;T[rc].mx=T[x].cov;T[lc].cov=T[x].cov;T[rc].cov=T[x].cov;T[lc].plu=0;T[rc].plu=0;}if(T[x].plu){T[lc].mx+=T[x].plu;T[rc].mx+=T[x].plu;T[lc].plu+=T[x].plu;T[rc].plu+=T[x].plu;}T[x].plu=0;T[x].cov=-1;
}
void pushup(int x)
{T[x].mx=max(T[x<<1].mx,T[x<<1|1].mx);
}
void build(int x,int l,int r)
{T[x].l=l,T[x].r=r;T[x].cov=-1;if(l==r){T[x].mx=a[npos[l]];return;}int mid=(l+r)>>1;build(x<<1,l,mid);build(x<<1|1,mid+1,r);pushup(x);
}
void add(int f,int t)
{to[++cnt]=t;nxt[cnt]=lj[f];lj[f]=cnt;
}
void dfs1(int x)
{dep[x]=dep[fa[x]]+1;s[x]=1;for(int i=lj[x];i;i=nxt[i])if(to[i]!=fa[x]){fa[to[i]]=x;dfs1(to[i]);s[x]+=s[to[i]];if(s[to[i]]>s[son[x]]) son[x]=to[i];}
}
void dfs2(int x)
{pos[x]=++cnt;npos[cnt]=x;if(son[fa[x]]==x) top[x]=top[fa[x]];else top[x]=x;if(son[x]) dfs2(son[x]);for(int i=lj[x];i;i=nxt[i])if(to[i]!=son[x]&&to[i]!=fa[x]) dfs2(to[i]);
}
int Query(int x,int l,int r)
{if(T[x].l==l&&T[x].r==r) return T[x].mx;if(T[x].plu||T[x].cov>=0) pushdown(x);int mid=(T[x].l+T[x].r)>>1;if(r<=mid) return Query(x<<1,l,r);else if(l>mid) return Query(x<<1|1,l,r);else return max(Query(x<<1,l,mid),Query(x<<1|1,mid+1,r));
}
void change(int x,int k,int v)
{if(T[x].l==T[x].r){T[x].mx=v;return;}int mid=(T[x].l+T[x].r)>>1;if(T[x].plu||T[x].cov>=0) pushdown(x);if(k<=mid) change(x<<1,k,v);else change(x<<1|1,k,v);pushup(x);
}
void Modify(int x,int l,int r,int v,int p)
{if(T[x].l==l&&T[x].r==r){if(p==1) T[x].plu+=v,T[x].mx+=v;if(p==2) T[x].cov=v,T[x].plu=0,T[x].mx=v;return;}if(T[x].plu||T[x].cov>=0) pushdown(x);int mid=(T[x].l+T[x].r)>>1;if(r<=mid) Modify(x<<1,l,r,v,p);else if(l>mid) Modify(x<<1|1,l,r,v,p);else Modify(x<<1,l,mid,v,p),Modify(x<<1|1,mid+1,r,v,p);pushup(x);
}
int getQuery(int x,int y)
{int fx=top[x],fy=top[y],mx=0;while(fx!=fy){if(dep[fx]<dep[fy]) swap(x,y),swap(fx,fy);mx=max(mx,Query(1,pos[fx],pos[x]));x=fa[fx];fx=top[x];}if(dep[x]<dep[y]) swap(x,y);if(x!=y) mx=max(mx,Query(1,pos[y]+1,pos[x]));return mx;
}
void getModi(int x,int y,int v,int p)
{int fx=top[x],fy=top[y];while(fx!=fy){if(dep[fx]<dep[fy]) swap(x,y),swap(fx,fy);Modify(1,pos[fx],pos[x],v,p);x=fa[fx];fx=top[x];}if(dep[x]<dep[y]) swap(x,y);if(x!=y) Modify(1,pos[y]+1,pos[x],v,p);
}
int main()
{scanf("%d",&n);for(int i=1;i<n;i++){scanf("%d%d%d",&ff[i],&tt[i],&pp[i]);add(ff[i],tt[i]),add(tt[i],ff[i]);}cnt=0;dfs1(1);dfs2(1);for(int i=1;i<n;i++){if(dep[ff[i]]<dep[tt[i]]) a[tt[i]]=pp[i],dian[i]=tt[i];else a[ff[i]]=pp[i],dian[i]=ff[i];}build(1,1,n);char c[10];int x,y,h;while(scanf("%s",c)!=EOF){if(c[0]=='S') return 0;scanf("%d%d",&x,&y);if(c[1]=='a') printf("%d\n",getQuery(x,y));else if(c[1]=='h') change(1,pos[dian[x]],y);else{scanf("%d",&h);if(c[0]=='A') getModi(x,y,h,1);if(c[0]=='C') getModi(x,y,h,2);}}
}</span></span>
</span>

Description

毛毛虫经过及时的变形，最终逃过的一劫，离开了菜妈的菜园。毛毛虫经过千山万水，历尽千辛万苦，最后来到了小小的绍兴一中的校园里。爬啊爬~爬啊爬~~毛毛虫爬到了一颗小小的“毛景树”下面，发现树上长着他最爱吃的毛毛果~~~ “毛景树”上有N个节点和N-1条树枝，但节点上是没有毛毛果的，毛毛果都是长在树枝上的。但是这棵“毛景树”有着神奇的魔力，他能改变树枝上毛毛果的个数：  Change k w：将第k条树枝上毛毛果的个数改变为w个。  Cover u v w：将节点u与节点v之间的树枝上毛毛果的个数都改变为w个。  Add u v w：将节点u与节点v之间的树枝上毛毛果的个数都增加w个。由于毛毛虫很贪，于是他会有如下询问：  Max u v：询问节点u与节点v之间树枝上毛毛果个数最多有多少个。

Input

第一行一个正整数N。接下来N-1行，每行三个正整数Ui，Vi和Wi，第i+1行描述第i条树枝。表示第i条树枝连接节点Ui和节点Vi，树枝上有Wi个毛毛果。接下来是操作和询问，以“Stop”结束。

Output

对于毛毛虫的每个询问操作，输出一个答案。

Sample Input

4
1 2 8
1 3 7
3 4 9
Max 2 4
Cover 2 4 5
Add 1 4 10
Change 1 16
Max 2 4
Stop

Sample Output

9
16

【Data Range】
1<=N<=100,000，操作+询问数目不超过100,000。
保证在任意时刻，所有树枝上毛毛果的个数都不会超过10^9个。