bzoj4999 This Problem Is Too Simple！

Description

给您一颗树，每个节点有个初始值。

现在支持以下两种操作：

1. C i x(0<=x<2^31) 表示将i节点的值改为x。

2. Q i j x(0<=x<2^31) 表示询问i节点到j节点的路径上有多少个值为x的节点。

Input

第一行有两个整数N,Q（1 ≤N≤ 100,000；1 ≤Q≤ 200,000），分别表示节点个数和操作个数。

下面一行N个整数，表示初始时每个节点的初始值。

接下来N-1行，每行两个整数x,y，表示x节点与y节点之间有边直接相连（描述一颗树）。

接下来Q行，每行表示一个操作，操作的描述已经在题目描述中给出。

Output

对于每个Q输出单独一行表示所求的答案。

Sample Input

5 6
10 20 30 40 50
1 2
1 3
3 4
3 5
Q 2 3 40
C 1 40
Q 2 3 40
Q 4 5 30
C 3 10
Q 4 5 30

Sample Output

0
1
1
0

正解：树链剖分+线段树。

傻逼题，直接对于每一个权值建一棵线段树，然后树剖扣路径即可。

 #include <bits/stdc++.h>
 #define il inline
 #define RG register
 #define ll long long
 #define lb(x) (x & -x)
 #define N (300010)

 using namespace std;

 struct edge{ int nt,to; }g[N<<];
 struct data{ int op,x,y,v; }q[N];

 int head[N],top[N],fa[N],son[N],dfn[N],dep[N],sz[N],n,num,cnt;
 int sum[*N],ls[*N],rs[*N],rt[N],hsh[N],a[N],tot,Sz,Q;
 char ch[];

 il int gi(){
   RG int x=,q=; RG char ch=getchar();
   while ((ch<'' || ch>'') && ch!='-') ch=getchar();
   if (ch=='-') q=-,ch=getchar();
   while (ch>='' && ch<='') x=x*+ch-,ch=getchar();
   return q*x;
 }

 il void insert(RG int from,RG int to){
   g[++num]=(edge){head[from],to},head[from]=num; return;
 }

 il void dfs1(RG int x,RG int p){
   fa[x]=p,dep[x]=dep[p]+,sz[x]=; RG int v;
   for (RG int i=head[x];i;i=g[i].nt){
     v=g[i].to; if (v==p) continue;
     dfs1(v,x),sz[x]+=sz[v];
     if (sz[son[x]]<=sz[v]) son[x]=v;
   }
   return;
 }

 il void dfs2(RG int x,RG int p,RG int anc){
   top[x]=anc,dfn[x]=++cnt; RG int v;
   if (son[x]) dfs2(son[x],x,anc);
   for (RG int i=head[x];i;i=g[i].nt){
     v=g[i].to; if (v==p || v==son[x]) continue;
     dfs2(v,x,v);
   }
   return;
 }

 il void update(RG int &x,RG int l,RG int r,RG int p,RG int v){
   if (!x) x=++Sz; if (l==r){ sum[x]+=v; return; } RG int mid=(l+r)>>;
   p<=mid ? update(ls[x],l,mid,p,v) : update(rs[x],mid+,r,p,v);
   sum[x]=sum[ls[x]]+sum[rs[x]]; return;
 }

 il int query(RG int x,RG int l,RG int r,RG int xl,RG int xr){
   if (xl<=l && r<=xr) return sum[x]; RG int mid=(l+r)>>;
   if (xr<=mid) return query(ls[x],l,mid,xl,xr);
   else if (xl>mid) return query(rs[x],mid+,r,xl,xr);
   else return query(ls[x],l,mid,xl,mid)+query(rs[x],mid+,r,mid+,xr);
 }

 il int Query(RG int u,RG int v,RG int k){
   RG int ans=;
   while (top[u]!=top[v]){
     if (dep[top[u]]<dep[top[v]]) swap(u,v);
     ans+=query(rt[k],,n,dfn[top[u]],dfn[u]),u=fa[top[u]];
   }
   if (dep[u]>dep[v]) swap(u,v);
   ans+=query(rt[k],,n,dfn[u],dfn[v]); return ans;
 }

 int main(){
 #ifndef ONLINE_JUDGE
   freopen("simple.in","r",stdin);
   freopen("simple.out","w",stdout);
 #endif
   n=gi(),Q=gi();
   for (RG int i=;i<=n;++i) a[i]=gi(),hsh[++tot]=a[i];
   for (RG int i=,x,y;i<n;++i) x=gi(),y=gi(),insert(x,y),insert(y,x);
   dfs1(,),dfs2(,,);
   for (RG int i=;i<=Q;++i){
     scanf("%s",ch);
     if (ch[]=='C') q[i].op=,q[i].x=gi(); else q[i].x=gi(),q[i].y=gi();
     q[i].v=gi(),hsh[++tot]=q[i].v;
   }
   sort(hsh+,hsh+tot+),tot=unique(hsh+,hsh+tot+)-hsh-;
   for (RG int i=;i<=n;++i) a[i]=lower_bound(hsh+,hsh+tot+,a[i])-hsh,update(rt[a[i]],,n,dfn[i],);
   for (RG int i=;i<=Q;++i){
     q[i].v=lower_bound(hsh+,hsh+tot+,q[i].v)-hsh;
     if (q[i].op) update(rt[a[q[i].x]],,n,dfn[q[i].x],-),update(rt[a[q[i].x]=q[i].v],,n,dfn[q[i].x],);
     else printf("%d\n",Query(q[i].x,q[i].y,q[i].v));
   }
   return ;
 }