【GDOI2103模拟3.17】数树数

Description

给出一个n个节点的树，每个点有权值。给出m次操作，每次操作修改一个点的权值或者询问从x到y的路径上有多少个权值为z的点。
n<=10^5,m<=2*10^5,权值<2^31

Solution

话说2103年的题真的很鬼畜。
链剖likeSDOI2012旅行，还更简单。好像有把权值重复赋值的操作。
注意细节。（哪道题不用(=@__@=)？）

对了，还要离散化。

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define rep(i,a) for(int i=last[a];i;i=next[i])
#define N 100005 
using namespace std;
map<int,int> ha;
struct note{int ty,x,y,id;}ask[N*6];
int n,m,a[N],d[N],size[N],son[N],ans[N*2],num,tot,l,x,y,z;
int last[N*3],t[N*6],next[N*6],h[N*3],fa[N],top[N],w[N],tree[N*3];
bool bz[N*2];
char ch[2];
void add(int x,int y) {
    t[++l]=y;if (!last[x]) last[x]=l;
    if (h[x]) next[h[x]]=l;h[x]=l;
}
void dfs(int x,int y) {
    d[x]=d[y]+1;fa[x]=y;size[x]=1;int k=0;
    rep(i,x) if (t[i]!=y) {
        dfs(t[i],x);size[x]+=size[t[i]];
        if (size[t[i]]>k) k=size[t[i]],son[x]=t[i];
    }
}
void make(int x,int y) {
    w[x]=++tot;top[x]=y;
    if (!son[x]) return;
    make(son[x],y);
    rep(i,x) if (t[i]!=fa[x]&&t[i]!=son[x]) make(t[i],t[i]);
}
void change(int v,int l,int r,int x,int y) {
    if (l==r) {tree[v]+=y;return;}
    int m=(l+r)/2;
    if (x<=m) change(v*2,l,m,x,y);
    else change(v*2+1,m+1,r,x,y);
    tree[v]=tree[v*2]+tree[v*2+1];
}
int find(int v,int l,int r,int x,int y) {
    if (l==x&&r==y) return tree[v];
    int m=(l+r)/2;
    if (y<=m) return find(v*2,l,m,x,y);
    else if (x>m) return find(v*2+1,m+1,r,x,y);
    else return find(v*2,l,m,x,m)+find(v*2+1,m+1,r,m+1,y);
}
int query(int x,int y) {
    int f1=top[x],f2=top[y],ans=0;
    while (f1!=f2) {
        if (d[f1]<d[f2]) swap(f1,f2),swap(x,y);
        ans+=find(1,1,n,w[f1],w[x]);
        x=fa[f1];f1=top[x];
    }
    if (d[x]>d[y]) swap(x,y);
    ans+=find(1,1,n,w[x],w[y]);
    return ans;
}
void solve(int x) {
    rep(i,x) {
        int k=t[i];
        if (ask[k].ty) change(1,1,n,w[ask[k].x],ask[k].y);
        else ans[ask[k].id]=query(ask[k].x,ask[k].y);
    }
}
int main() {
    scanf("%d%d",&n,&m);
    fo(i,1,n) {
        scanf("%d",&a[i]);
        if (!ha[a[i]]) ha[a[i]]=++num;
        a[i]=ha[a[i]];
    }
    fo(i,1,n-1) scanf("%d%d",&x,&y),add(x,y),add(y,x);
    dfs(1,0);make(1,1);
    tot=l=0;
    memset(last,0,sizeof(last));memset(h,0,sizeof(h));
    memset(next,0,sizeof(next));memset(t,0,sizeof(t));
    fo(i,1,n) {
        ask[++tot].ty=1;ask[tot].x=i;ask[tot].y=1;
        add(a[i],tot);
    }
    fo(i,1,m) {
        scanf("%s",ch);
        if (ch[0]=='Q') {
            scanf("%d%d%d",&x,&y,&z);
            if (!ha[z]) ha[z]=++num;z=ha[z];
            ask[++tot].x=x;ask[tot].y=y;ask[tot].id=i;
            add(z,tot);bz[i]=1;
        } else {
            scanf("%d%d",&x,&y);
            if (!ha[y]) ha[y]=++num;y=ha[y];
            ask[++tot].ty=1;ask[tot].x=x;ask[tot].y=1;add(y,tot);
            ask[++tot].ty=1;ask[tot].x=x;ask[tot].y=-1;add(a[x],tot);
            a[x]=y;
        }
    }
    fo(i,1,n) {
        ask[++tot].ty=1;ask[tot].x=i;ask[tot].y=-1;
        add(a[i],tot);
    }
    fo(i,1,num) solve(i);
    fo(i,1,m) if (bz[i]) printf("%d\n",ans[i]);
}