【清华冬令营2018模拟】送你一棵圣诞树

Description

送你一棵n 个点的树, 树根为1. 一开始每个点上有一个1:::n 的颜色ci, 不同点颜色可以相同.
现在有q 次操作, 分为两种类型:
• 1 u l r: 询问子树u 中有多少种在l 到r 之间的颜色至少出现了一次
• 2 u c: 将u 的颜色修改为c
部分测试点要求强制在线.

Input

第一行三个整数n; q; t, 分别表示树的点数, 操作的个数和是否强制在线. t = 0 表示不强制在
线, t = 1 表示强制在线.
接下来一行n 个整数ci, 表示每个点的初始颜色.
接下来n �� 1 行, 每行两个整数ui; vi, 表示一条ui 到vi 的边.
接下来q 行, 每行四个或三个整数, 表示一个操作. 当t = 1 时, 需要对第一个数以外的其他数
异或上一次询问的答案lastans, 初始时lastans = 0.

Output

对于每个询问输出一行一个整数, 表示答案.

Sample 1

Input

5 5 0
5 5 2 5 5
5 1
2 5
4 2
3 5
1 2 2 3
2 5 1
1 1 1 5
2 3 2
1 3 1 5

Output

0
3
1

Sample 2

Input

5 5 1
4 1 1 5 4
5 1
3 5
2 3
4 3
2 5 4
2 2 2
1 3 1 5
2 1 2
1 1 2 7

Output

3
1

Solution

这题虽说是数据结构题，但是不很经典
比较经典的做法就是按照dfs序线段树或者树链剖分，这是比较好想的
但是这题如果这么想就容易想歪

首先对于每个颜色，开个set记录这个颜色的点的dfs序，也就是排好序了的
对一个颜色，每个点自己位置+1，set中dfs序相邻位置的lca-1，这样统计贡献时，直接子树权值和就行（画张图理解一下）
多个颜色就开树状数组套线段树就行了

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define N 101000
#define lowbit(x) (x&(-x))
using namespace std;
int a[N],last[N],next[N*2],to[N*2],n,q,zx,f[N][17],bz[N],tot=0,t[N*2],ans,dfn[N],low[N],deep[N],fdfn[N];
struct node{
    int sz,l,r;
}g[N*200];
set<int> c[N];
typedef set<int> :: iterator it;
void putin(int x,int y)
{
    next[++tot]=last[x];last[x]=tot;to[tot]=y;
}
void dg(int x)
{
    dfn[x]=++tot;fdfn[tot]=x;
    for(int i=last[x];i;i=next[i])
    {
        if(to[i]==f[x][0]) continue;
        f[to[i]][0]=x;deep[to[i]]=deep[x]+1;dg(to[i]);
    }
    low[x]=tot;
}
int lca(int x,int y)
{
    if(x==y||x==0||y==0||x==n+1||y==n+1) return 0;
    x=fdfn[x],y=fdfn[y];
    if(deep[x]<deep[y]) swap(x,y);
    fd(i,16,0) if(deep[f[x][i]]>=deep[y]) x=f[x][i];
    fd(i,16,0) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    if(x!=y) x=f[x][0];
    return dfn[x];
}
void change(int v,int i,int j,int x,int y)
{
    if(i==j){g[v].sz+=y;return;}
    int m=(i+j)/2;
    if(x<=m) g[v].l=g[v].l==0?++tot:g[v].l,change(g[v].l,i,m,x,y);
    else g[v].r=g[v].r==0?++tot:g[v].r,change(g[v].r,m+1,j,x,y);
    g[v].sz=g[g[v].l].sz+g[g[v].r].sz;
}
int find(int v,int i,int j,int x,int y)
{
    if(v==0) return 0;
    if(i==x&&j==y) return g[v].sz;
    int m=(i+j)/2;
    if(y<=m) return find(g[v].l,i,m,x,y);
    else if(x>m) return find(g[v].r,m+1,j,x,y);
         else return find(g[v].l,i,m,x,m)+find(g[v].r,m+1,j,m+1,y);
}
void ins(int x,int y,int z)
{
    for(;x<=n;x+=lowbit(x)) change(t[x],1,n,y,z);
}
int get(int x,int l,int r)
{
    int ans=0;
    for(;x;x-=lowbit(x)) ans+=find(t[x],1,n,l,r);
    return ans;
}
int main()
{
    freopen("xmastree1.in","r",stdin);
    freopen("xmastree1.out","w",stdout);
    scanf("%d%d%d",&n,&q,&zx);
    fo(i,1,n) scanf("%d",&a[i]);
    tot=0;
    fo(i,1,n-1)
    {
        int x,y;scanf("%d%d",&x,&y);
        putin(x,y);putin(y,x);
    }
    tot=0;deep[1]=1;dg(1);
    fo(j,1,16) fo(i,1,n) f[i][j]=f[f[i][j-1]][j-1];
    fo(i,1,n) t[i]=i,c[a[i]].insert(dfn[i]),c[i].insert(0),c[i].insert(n+1);
    tot=n;
    fo(i,1,n)
    {
        int m=c[i].size();
        if(m==2) continue;
        it pos=c[i].begin();
        pos++;
        int x=*pos;ins(i,x,1);
        fo(j,2,m-2)
        {
            pos++;
            ins(i,lca(x,*pos),-1);
            x=*pos;
            ins(i,x,1);
        }
    }
    while(q--)
    {
        int tag;
        scanf("%d",&tag);
        if(tag==1)
        {
            int x,l,r;scanf("%d%d%d",&x,&l,&r);
            x=x^(ans*zx),l=l^(ans*zx),r=r^(ans*zx);
            ans=get(r,dfn[x],low[x])-get(l-1,dfn[x],low[x]);
            printf("%d\n",ans);
        }
        if(tag==2)
        {
            int x,y;scanf("%d%d",&x,&y);
            x=x^(ans*zx),y=y^(ans*zx);
            it pos=c[a[x]].find(dfn[x]);
            int x1=*--pos;
            pos++;
            int x2=*++pos;
            int l1=lca(x1,dfn[x]);
            if(l1) ins(a[x],l1,1);
            l1=lca(x2,dfn[x]);
            if(l1) ins(a[x],l1,1);
            l1=lca(x1,x2);
            if(l1&&x1!=dfn[x]&&x2!=dfn[x]) ins(a[x],l1,-1);
            ins(a[x],dfn[x],-1);
            c[a[x]].erase(dfn[x]);

            a[x]=y;
            c[a[x]].insert(dfn[x]);
            pos=c[a[x]].find(dfn[x]);
            x1=*--pos;
            pos++;
            x2=*++pos;
            l1=lca(x1,dfn[x]);
            if(l1) ins(a[x],l1,-1);
            l1=lca(x2,dfn[x]);
            if(l1) ins(a[x],l1,-1);
            l1=lca(x1,x2);
            if(l1&&x1!=dfn[x]&&x2!=dfn[x]) ins(a[x],l1,1);
            ins(a[x],dfn[x],1);
        }
    }
}