luogu_P3313 [SDOI2014]旅行
阿新 • • 發佈:2018-12-17
傳送門
Solution
第二次學習可持久化線段樹
打了一道裸題來練習一下……
- 對於每個宗教都可以開一個主席樹
- 基礎操作
- 樹剖lca
Code
#include<bits/stdc++.h> #define ll long long #define max(a,b) ((a)>(b)?(a):(b)) #define min(a,b) ((a)<(b)?(a):(b)) inline int read() { int x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();} return x*f; } #define MN 100005 #define MM 8000005 int w[MN],c[MN],N; struct edge{int to,nex;}e[MN<<1]; int hr[MN],en; inline void ins(int f,int t) { e[++en]=(edge){t,hr[f]};hr[f]=en; e[++en]=(edge){f,hr[t]};hr[t]=en; } int fa[MN],dep[MN],mx[MN],siz[MN],pos[MN],top[MN]; void dfs1(int x,int f) { dep[x]=dep[fa[x]=f]+(siz[x]=1);register int i; for(i=hr[x];i;i=e[i].nex) if(f^e[i].to) dfs1(e[i].to,x),siz[e[i].to]>siz[mx[x]]?mx[x]=e[i].to:0,siz[x]+=siz[e[i].to]; } int dind=0; void dfs2(int x,int tp) { pos[x]=++dind;top[x]=tp;register int i; if(mx[x]) dfs2(mx[x],tp); for(i=hr[x];i;i=e[i].nex) if((fa[x]^e[i].to)&&(mx[x]^e[i].to)) dfs2(e[i].to,e[i].to); } #define mid ((l+r)>>1) int Max[MM],Sum[MM],ls[MM],rs[MM],rt[MN],sz; int query_m(int x,int l,int r,int a,int b) { if(!x) return 0; if(a==l&&r==b) return Max[x]; if(b<=mid) return query_m(ls[x],l,mid,a,b); if(a>mid) return query_m(rs[x],mid+1,r,a,b); return max(query_m(ls[x],l,mid,a,mid),query_m(rs[x],mid+1,r,mid+1,b)); } int query_s(int x,int l,int r,int a,int b) { if(!x) return 0; if(a==l&&r==b) return Sum[x]; if(b<=mid) return query_s(ls[x],l,mid,a,b); if(a>mid) return query_s(rs[x],mid+1,r,a,b); return query_s(ls[x],l,mid,a,mid)+query_s(rs[x],mid+1,r,mid+1,b); } inline void pushup(int x){Max[x]=max(Max[ls[x]],Max[rs[x]]);Sum[x]=Sum[ls[x]]+Sum[rs[x]];} void Modify(int &x,int l,int r,int a,int v) { if(!x) x=++sz; if(l==r) return (void)(Max[x]=Sum[x]=v); if(a<=mid) Modify(ls[x],l,mid,a,v); else Modify(rs[x],mid+1,r,a,v); pushup(x); } int QS(int root,int x,int y) { int res=0; while(top[x]^top[y]) { if(dep[top[x]]>dep[top[y]]) res+=query_s(root,1,N,pos[top[x]],pos[x]),x=fa[top[x]]; else res+=query_s(root,1,N,pos[top[y]],pos[y]),y=fa[top[y]]; } res+=query_s(root,1,N,min(pos[x],pos[y]),max(pos[x],pos[y])); return res; } int QM(int root,int x,int y) { int res=0; while(top[x]^top[y]) { if(dep[top[x]]>dep[top[y]]) res=max(res,query_m(root,1,N,pos[top[x]],pos[x])),x=fa[top[x]]; else res=max(res,query_m(root,1,N,pos[top[y]],pos[y])),y=fa[top[y]]; } res=max(res,query_m(root,1,N,min(pos[x],pos[y]),max(pos[x],pos[y]))); return res; } int main() { register int i,Q,x,y;register char opt[10]; N=read(),Q=read(); for(i=1;i<=N;++i) w[i]=read(),c[i]=read(); for(i=1;i<N;++i) x=read(),ins(x,read()); dfs1(1,0);dfs2(1,1); for(i=1;i<=N;++i) Modify(rt[c[i]],1,N,pos[i],w[i]); while(Q--) { scanf("%s",opt+1);x=read();y=read(); if(opt[1]=='C'&&opt[2]=='C') Modify(rt[c[x]],1,N,pos[x],0),Modify(rt[c[x]=y],1,N,pos[x],w[x]); if(opt[1]=='C'&&opt[2]=='W') Modify(rt[c[x]],1,N,pos[x],w[x]=y); if(opt[1]=='Q'&&opt[2]=='S') printf("%d\n",QS(rt[c[y]],x,y)); if(opt[1]=='Q'&&opt[2]=='M') printf("%d\n",QM(rt[c[y]],x,y)); } return 0; }
Blog來自PaperCloud,未經允許,請勿轉載,TKS!