GDOI2016模拟8.14数树数

题目
给定一棵N 个节点的树，标号从1~N。每个点有一个权值。要求维护两种操作：
1. C i x(0<=x<2^31) 表示将i 点权值变为x
2. Q i j x(0<=x<2^31) 表示询问i 到j 的路径上有多少个值为x 的节点

比较简单的想法就是按权值建N+Q棵可持久化线段树，第i棵线段树里记录权值为a[i]的节点的区间和，节点要用链剖后的dfs序来做编号，修改时就是将原权值所在线段树中的点删除，在新权值线段树里插入，查询就沿重链往上跳，先跳深度大的，然后在线段树里查询就行了。

时间复杂度：（N+Q）(log(N+Q))^2
空间复杂度：（N+Q）(log(N+Q))

贴代码

#include<iostream>#include<cstdio>#include<algorithm>#define N 300001using namespace std;int n,m,sum,ans;int fa[N],q[N][4],g[N],a[N+N][2],f[N*36][3],bz[N],h[N],v[N],b[N][2],c[N],w[N];void ins(int x,int y){    static int sum=0;    a[++sum][0]=y,a[sum][1]=g[x],g[x]=sum;}void init(){    static char x;    static int y,z;    scanf("%d %d",&n,&m);    for (int i=1;i<=n;i++)        scanf("%d",&v[i]),c[++c[0]]=v[i];    for (int i=1;i<n;i++)        scanf("%d %d",&y,&z),ins(y,z),ins(z,y);    for (int i=1;i<=m;i++){        scanf(" %c %d %d",&x,&q[i][1],&q[i][2]);        if (x=='Q')            scanf("%d",&q[i][3]),q[i][0]=1;        else            c[++c[0]]=q[i][2];    }}void dfs(int x){    for (int i=g[x];i;i=a[i][1])        if (a[i][0]!=fa[x]){            fa[a[i][0]]=x;            dfs(a[i][0]);            if (b[x][0]<b[a[i][0]][0]+1)                b[x][0]=b[a[i][0]][0]+1,b[x][1]=a[i][0];        }}void dfs1(int x){    static int sum=0;    bz[x]=++sum;    if (b[x][1])        h[b[x][1]]=h[x],dfs1(b[x][1]);    for (int i=g[x];i;i=a[i][1])        if (a[i][0]!=fa[x]&&a[i][0]!=b[x][1])            h[a[i][0]]=a[i][0],dfs1(a[i][0]);}void change(int l,int r,int s,int ll,int z){    f[s][2]+=z;    if (l==r)return;    static int ss;    if ((ss=(l+r)/2)>=ll){        if (!f[s][0])f[s][0]=++sum;        change(l,ss,f[s][0],ll,z);    }else{        if (!f[s][1])f[s][1]=++sum;        change(ss+1,r,f[s][1],ll,z);    }}int er(int x){    static int l,r,mid;    l=1,r=c[0];    while (l<=r)if (c[mid=(l+r)/2]<x)l=mid+1;else r=mid-1;    if (l<1)l=1;    if (l>c[0])l=c[0];    while (c[l]<x&&l<c[0])l++;    while (c[l]>x&&l>1)l--;    return l;}void pre(){    static int x;    sort(c+1,c+c[0]+1);    x=c[0],c[0]=1;    for (int i=2;i<=x;i++)        if (c[c[0]]!=c[i])c[++c[0]]=c[i];    dfs(1);    h[1]=1;    dfs1(1);    for (int i=1;i<=c[0];i++)        w[i]=++sum;    for (int i=1;i<=n;i++)        change(1,n,w[er(v[i])],bz[i],1);}void find(int l,int r,int s,int ll,int rr){    if (!f[s][2])return;    if (rr<l||r<ll)return;    if (ll<=l&&r<=rr){        ans+=f[s][2];        return;    }    find(l,(l+r)/2,f[s][0],ll,rr),find((l+r)/2+1,r,f[s][1],ll,rr);}void up(int x,int y,int z){    static int ss;    ss=z;    z=er(z);    if (c[z]!=ss)return;    while (h[x]!=h[y]){        if (bz[h[x]]<bz[h[y]])swap(x,y);        find(1,n,w[z],bz[h[x]],bz[x]);        x=fa[h[x]];    }    if (bz[x]<bz[y])swap(x,y);    find(1,n,w[z],bz[y],bz[x]);}void work(){    for (int i=1;i<=m;i++)        if (q[i][0]){            ans=0;            up(q[i][1],q[i][2],q[i][3]);            printf("%d\n",ans);        }else{            change(1,n,w[er(v[q[i][1]])],bz[q[i][1]],-1);            v[q[i][1]]=q[i][2];            change(1,n,w[er(q[i][2])],bz[q[i][1]],1);        }}int main(){    init();    pre();    work();    return 0;}