标签:read 题解 char define tput next 个数 val const
【Data Range】
1<=N<=100,000,操作+询问数目不超过100,000。
保证在任意时刻,所有树枝上毛毛果的个数都不会超过10^9个。
题解:树剖+线段树裸题,不过注意是边权,把边权转换成点权,去掉最上面的点即可。还要注意一下cover和add两个标签的处理顺序。
#include <stdio.h> #include <string.h> #include <iostream> #define lson x<<1 #define rson x<<1|1 using namespace std; const int maxn=150010; int n,cnt,tot; int s[maxn<<2],p[maxn],tc[maxn<<2],ts[maxn<<2]; int to[maxn<<1],next[maxn<<1],val[maxn<<1],v[maxn],u[maxn],fa[maxn],head[maxn]; int deep[maxn],size[maxn],son[maxn],top[maxn]; char str[10]; int readin() { int ret=0,sig=1; char gc; while(gc<‘0‘||gc>‘9‘) sig=(gc==‘-‘)?-1:1,gc=getchar(); while(gc>=‘0‘&&gc<=‘9‘) ret=ret*10+gc-‘0‘,gc=getchar(); return ret*sig; } void add(int a,int b,int c) { to[cnt]=b; val[cnt]=c; next[cnt]=head[a]; head[a]=cnt++; } void dfs1(int x) { size[x]=1; for(int i=head[x];i!=-1;i=next[i]) { if(to[i]!=fa[x]) { fa[to[i]]=x; deep[to[i]]=deep[x]+1; u[to[i]]=val[i]; dfs1(to[i]); size[x]+=size[to[i]]; if(size[to[i]]>size[son[x]]) son[x]=to[i]; } } } void dfs2(int x,int tp) { top[x]=tp; p[x]=++tot; v[p[x]]=u[x]; if(son[x]) dfs2(son[x],tp); for(int i=head[x];i!=-1;i=next[i]) if(to[i]!=son[x]&&to[i]!=fa[x]) dfs2(to[i],to[i]); } void pushup(int x) { s[x]=max(s[lson],s[rson]); } void pushdown(int x) { if(tc[x]!=-1) //注意顺序 { s[lson]=s[rson]=tc[lson]=tc[rson]=tc[x]; ts[lson]=ts[rson]=0; tc[x]=-1; } if(ts[x]) { s[lson]+=ts[x],s[rson]+=ts[x],ts[lson]+=ts[x],ts[rson]+=ts[x]; ts[x]=0; } } void build(int l,int r,int x) { if(l==r) { s[x]=v[l]; return ; } int mid=l+r>>1; build(l,mid,lson),build(mid+1,r,rson); pushup(x); } int query(int l,int r,int x,int a,int b) { pushdown(x); if(a<=l&&r<=b) return s[x]; int mid=l+r>>1; if(b<=mid) return query(l,mid,lson,a,b); if(a>mid) return query(mid+1,r,rson,a,b); return max(query(l,mid,lson,a,b),query(mid+1,r,rson,a,b)); } void upcover(int l,int r,int x,int a,int b,int c) { pushdown(x); if(a<=l&&r<=b) { s[x]=tc[x]=c; return ; } int mid=l+r>>1; if(b<=mid) upcover(l,mid,lson,a,b,c); else if(a>mid) upcover(mid+1,r,rson,a,b,c); else upcover(l,mid,lson,a,b,c),upcover(mid+1,r,rson,a,b,c); pushup(x); } void updata(int l,int r,int x,int y,int c) { if(l==r) { s[x]=c; return ; } pushdown(x); int mid=l+r>>1; if(y<=mid) updata(l,mid,lson,y,c); else updata(mid+1,r,rson,y,c); pushup(x); } void upadd(int l,int r,int x,int a,int b,int c) { pushdown(x); if(a<=l&&r<=b) { s[x]+=c,ts[x]+=c; return ; } int mid=l+r>>1; if(b<=mid) upadd(l,mid,lson,a,b,c); else if(a>mid) upadd(mid+1,r,rson,a,b,c); else upadd(l,mid,lson,a,b,c),upadd(mid+1,r,rson,a,b,c); pushup(x); } void getmax() { int x=readin(),y=readin(),ans=0; while(top[x]!=top[y]) { if(deep[top[x]]<deep[top[y]]) swap(x,y); ans=max(ans,query(1,n,1,p[top[x]],p[x])); x=fa[top[x]]; } if(deep[x]>deep[y]) swap(x,y); if(x!=y) ans=max(ans,query(1,n,1,p[x]+1,p[y])); //p[x]要+1,取下面那一个点 printf("%d\n",ans); } void Cover() { int x=readin(),y=readin(),z=readin(); while(top[x]!=top[y]) { if(deep[top[x]]<deep[top[y]]) swap(x,y); upcover(1,n,1,p[top[x]],p[x],z); x=fa[top[x]]; } if(deep[x]>deep[y]) swap(x,y); if(x!=y) upcover(1,n,1,p[x]+1,p[y],z); } void Change() { int x=readin(),y=readin(); updata(1,n,1,max(p[to[x*2-2]],p[to[x*2-1]]),y); } void Add() { int x=readin(),y=readin(),z=readin(); while(top[x]!=top[y]) { if(deep[top[x]]<deep[top[y]]) swap(x,y); upadd(1,n,1,p[top[x]],p[x],z); x=fa[top[x]]; } if(deep[x]>deep[y]) swap(x,y); if(x!=y) upadd(1,n,1,p[x]+1,p[y],z); } int main() { n=readin(); int i,a,b,c; memset(head,-1,sizeof(head)); memset(tc,-1,sizeof(tc)); for(i=1;i<n;i++) { a=readin(),b=readin(),c=readin(); add(a,b,c),add(b,a,c); } deep[1]=1; dfs1(1); dfs2(1,1); build(1,n,1); while(scanf("%s",str)) { switch(str[1]) { case ‘a‘:getmax(); break; case ‘o‘:Cover(); break; case ‘d‘:Add(); break; case ‘h‘:Change(); break; case ‘t‘:return 0; } } }
标签:read 题解 char define tput next 个数 val const
原文地址:http://www.cnblogs.com/CQzhangyu/p/6184440.html