[bzoj4940][莫队算法]这是我自己的发明

Description

给一个树，n 个点，有点权，初始根是 1。 m 个操作，每次操作：
1. 将树根换为 x。
2. 给出两个点 x，y，从 x 的子树中选每一个点，y 的子树中选每一个点，如果两个点点权相等，ans++，求 ans。

Input

第一行两个数表示 n，m。 第二行 n 个数，表示每个点的点权 a[i]。 之后 n ? 1 行 , 每行两个数 x , y , 表示一条边
之后 m 行，每行为 1 x 或者 2 x y。 1 x，表示将根变成 x点。 2 x y，表示查询 x 点的子树与 y 点的子树。 n
<= 100000 , m <= 500000 , 1 <= a[i] <= 1000000000

Output

对于每个询问，输出一个数表示答案。

Sample Input

5 5

1 2 3 4 5

1 2

1 3

3 4

3 5

2 4 5

2 1 5

2 3 5

1 5

2 4 5

Sample Output

0

1

1

1

题解

这题其实就是3083与5016的结合版
你可以把一个询问拆成九个搞
然后卡了半小时常数。。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
inline int read()
{
    int f=1,x=0;char ch=getchar();
    while(ch<'0' || ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
struct LSnode{int y,p;}w[110000];int col[110000];
bool cmp(LSnode n1,LSnode n2){return n1.y<n2.y;}
struct node{int x,y,next;}a[210000];int len,last[110000];
inline void ins(int x,int y){++len;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;}
int dep[110000],bin[25],fa[110000][25];
int in[110000],ot[110000],dfn;
inline void pre_tree_node(int x)
{
    in[x]=++dfn;
    for(register int i=1;bin[i]<=dep[x];++i)fa[x][i]=fa[fa[x][i-1]][i-1];
    for(register int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa[x][0])
        {
            fa[y][0]=x;dep[y]=dep[x]+1;
            pre_tree_node(y);
        }
    }
    ot[x]=dfn;
}
int Root;
inline int lca(int x,int y)
{
    if(dep[x]<dep[y])swap(x,y);
    for(register int i=17;i>=0;--i)if(bin[i]<=dep[x] && dep[fa[x][i]]>=dep[y])x=fa[x][i];
    if(x==y)return x;
    for(register int i=17;i>=0;--i)if(bin[i]<=dep[x] && fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
inline int go(int x,int h)
{
    for(register int i=17;i>=0;--i)if(bin[i]<=h)x=fa[x][i],h-=bin[i];
    return x;
}
struct ask{int l,r,op,Pgo;}A[4500010];int cnt;
inline void add(int l,int r,int op,int Pgo){cnt++;A[cnt].l=l;A[cnt].r=r;A[cnt].op=op;A[cnt].Pgo=Pgo;}
int pos[110000],block;
bool cmpx(ask n1,ask n2){return pos[n1.l]!=pos[n2.l]?pos[n1.l]<pos[n2.l]:n1.r<n2.r;}
int n,m;
int answer[510000],ans;
int cx[110000],cy[110000];
inline void addl(int now){ans+=cy[col[now]];++cx[col[now]];}
inline void dell(int now){ans-=cy[col[now]];--cx[col[now]];}
inline void addr(int now){ans+=cx[col[now]];++cy[col[now]];}
inline void delr(int now){ans-=cx[col[now]];--cy[col[now]];}
bool v[510000];
inline void write(int x)
{
    if(!x) return (void)puts("0");
    if(x<0) putchar('-'),x=-x;
    static short s[25],t;
    while(x) s[++t]=x%10,x/=10; 
    while(t) putchar('0'+s[t--]);
    putchar('\n');
}
inline int fd(int x)
{
    if(dep[x]>=dep[Root])return Root;
    if(in[x]<=in[Root] && ot[x]>=in[Root])return x;
    return Root;
}
int main()
{
    bin[0]=1;for(register int i=1;i<=20;++i)bin[i]=bin[i-1]<<1;
    n=read();m=read();block=sqrt(n);
    for(register int i=1;i<=n;++i)w[i].y=read(),w[i].p=i,pos[i]=(i-1)/block+1;
    len=0;memset(last,0,sizeof(last));
    for(register int i=1;i<n;++i)
    {
        register int x=read(),y=read();
        ins(x,y);ins(y,x);
    }
    fa[1][0]=0;dep[1]=1;pre_tree_node(1);   
    sort(w+1,w+1+n,cmp);
    int tt=1;col[in[w[1].p]]=1;
    for(register int i=2;i<=n;++i)
    {
        if(w[i].y!=w[i-1].y)tt++;
        col[in[w[i].p]]=tt;
    }
    Root=1;
    memset(v,false,sizeof(v));
    for(register int i=1;i<=m;++i)
    {
        register int op=read(),x=read(),y;
        if(op==1)Root=x;
        else
        {
            v[i]=true;y=read();
            if(Root==x && Root==y){add(n,n,1,i);continue;}
            if(Root==y && Root!=x)swap(x,y);
            //register int tx=lca(x,Root),ty=lca(y,Root);
            int tx=fd(x),ty=fd(y);
            if(Root==x && Root!=y)
            {
                if(ty==y)
                {
                    y=go(Root,dep[Root]-dep[y]-1);
                    add(n,in[y]-1,1,i);
                    add(n,n,1,i);
                    add(n,ot[y],-1,i);
                    continue;
                }
                else
                {
                    add(n,ot[y],1,i);
                    add(n,in[y]-1,-1,i);
                    continue;
                }
            }
            if(tx==x && ty!=y)swap(x,y),swap(tx,ty);//x要换根，y不用 
            if(tx!=x && ty!=y)
            {
                add(ot[x],ot[y],1,i);
                add(ot[x],in[y]-1,-1,i);
                add(in[x]-1,ot[y],-1,i);
                add(in[x]-1,in[y]-1,1,i);
            }
            else if(tx!=x && ty==y)
            {
                y=go(Root,dep[Root]-dep[y]-1);
                add(ot[x],in[y]-1,1,i);
                add(ot[x],n,1,i);
                add(ot[x],ot[y],-1,i);
                add(in[x]-1,in[y]-1,-1,i);
                add(in[x]-1,n,-1,i);
                add(in[x]-1,ot[y],1,i);
            }
            else
            {
                x=go(Root,dep[Root]-dep[x]-1);
                y=go(Root,dep[Root]-dep[y]-1);
                add(in[x]-1,in[y]-1,1,i);
                add(in[x]-1,n,1,i);
                add(in[x]-1,ot[y],-1,i);
                add(n,in[y]-1,1,i);
                add(n,n,1,i);
                add(n,ot[y],-1,i);
                add(ot[x],in[y]-1,-1,i);
                add(ot[x],n,-1,i);
                add(ot[x],ot[y],1,i);
            }
        }
    }
    sort(A+1,A+1+cnt,cmpx);
    ans=0;
    for(register int i=1;i<=A[1].l;++i)++cx[col[i]];
    for(register int i=1;i<=A[1].r;++i)ans+=cx[col[i]],++cy[col[i]];
    answer[A[1].Pgo]+=ans*A[1].op;
    int l=A[1].l,r=A[1].r;
    for(register int i=2;i<=cnt;++i)
    {
        while(l<A[i].l)addl(++l);
        while(l>A[i].l)dell(l--);
        while(r<A[i].r)addr(++r);
        while(r>A[i].r)delr(r--);
        answer[A[i].Pgo]+=ans*A[i].op;
    }
    for(register int i=1;i<=m;++i)if(v[i])write(answer[i]);
    return 0;
}