标签:一个 struct int ext end main printf 线段 数组
题解:
这一道题目和模板的差别在于这一刀题目还有一个信仰
如果没有信仰,那么就是模板
然后考虑对于每一个信仰,我们都建立一颗线段树
注意要动态开点
所以要数组多开大一些
代码:
#include<bits/stdc++.h> using namespace std; const int N=200005; int n,m,cnt,tot,al,roots[N],pre[N],c[N],w[N],top[N]; int x,y,z,fa[N],son[N],siz[N],dep[N],L[N],first[N]; char ch; struct edge { int u,v,next; }e[N]; struct node { int maxn,sum,lson,rson; }T[N*10]; void pushup(int now) { T[now].maxn=max(T[T[now].lson].maxn,T[T[now].rson].maxn); T[now].sum=T[T[now].lson].sum+T[T[now].rson].sum; } void add(int x,int y) { e[++tot].u=x; e[tot].v=y; e[tot].next=first[x]; first[x]=tot; } void dfs1(int now) { siz[now]=1; for (int i=first[now];i;i=e[i].next) if (e[i].v!=fa[now]) { fa[e[i].v]=now; dep[e[i].v]=dep[now]+1; dfs1(e[i].v); if (siz[e[i].v]>siz[son[now]]) son[now]=e[i].v; siz[now]+=siz[e[i].v]; } } void dfs2(int now,int tp) { top[now]=tp;pre[++cnt]=now;L[now]=cnt; if (son[now]) dfs2(son[now],tp); for (int i=first[now];i;i=e[i].next) if (e[i].v!=fa[now]&&e[i].v!=son[now]) dfs2(e[i].v,e[i].v); } void update(int &now,int begin,int end,int pos,int num) { if (!now) now=++al; if (begin==end) {T[now].maxn=T[now].sum=num;return;} int mid=(begin+end)>>1; if (pos<=mid) update(T[now].lson,begin,mid,pos,num); else update(T[now].rson,mid+1,end,pos,num); pushup(now); } int get_sum(int now,int begin,int end,int l,int r) { if (l<=begin&&end<=r) return T[now].sum; int mid=(begin+end)>>1,ans=0; if (mid>=l) ans+=get_sum(T[now].lson,begin,mid,l,r); if (mid<r) ans+=get_sum(T[now].rson,mid+1,end,l,r); return ans; } int get_max(int now,int begin,int end,int l,int r) { if (l<=begin&&end<=r) return T[now].maxn; int mid=(begin+end)>>1,ans=0; if (mid>=l) ans=max(ans,get_max(T[now].lson,begin,mid,l,r)); if (mid<r) ans=max(ans,get_max(T[now].rson,mid+1,end,l,r)); return ans; } int find(int color,int l,int r) { int f1=top[l],f2=top[r],ans=0; while (f1!=f2) { if (dep[f1]<dep[f2]) swap(f1,f2),swap(l,r); if (ch==‘S‘) ans+=get_sum(roots[color],1,cnt,L[f1],L[l]); else ans=max(ans,get_max(roots[color],1,cnt,L[f1],L[l])); l=fa[f1];f1=top[l]; } if (dep[l]>dep[r]) swap(l,r); if (ch==‘S‘) ans+=get_sum(roots[color],1,cnt,L[l],L[r]); else ans=max(ans,get_max(roots[color],1,cnt,L[l],L[r])); return ans; } int main() { scanf("%d%d",&n,&m); for (int i=1;i<=n;i++)scanf("%d%d",&w[i],&c[i]); for (int i=1;i<n;i++)scanf("%d%d",&x,&y),add(x,y),add(y,x); dfs1(1);dfs2(1,1); for (int i=1;i<=n;i++)update(roots[c[i]],1,cnt,L[i],w[i]); while (m--) { ch=getchar(); while (ch!=‘C‘&&ch!=‘Q‘) ch=getchar(); if (ch==‘C‘) { ch=getchar(); scanf("%d%d",&x,&y); if (ch==‘C‘) { update(roots[c[x]],1,n,L[x],0); c[x]=y; update(roots[c[x]],1,n,L[x],w[x]); } else update(roots[c[x]],1,n,L[x],y),w[x]=y; } else { ch=getchar(); scanf("%d%d",&x,&y); printf("%d\n",find(c[x],x,y)); } } }
标签:一个 struct int ext end main printf 线段 数组
原文地址:http://www.cnblogs.com/xuanyiming/p/7994468.html