[Ynoi2016]这是我自己的发明

大概就是BZOJ5016[Snoi2017]一个简单的询问+BZOJ3083遥远的国度吧

换根就像遥远的国度一样讨论
然后像一个简单的询问一样差分
跑莫队即可

#include<bits/stdc++.h>

using namespace std;

#define gc c=getchar()
#define r(x) read(x)
#define ll long long

template<typename T>
inline void read(T&x){
    x=0;T k=1;char gc;
    while(!isdigit(c)){if(c=='-')k=-1;gc;}
    while(isdigit(c)){x=x*10+c-'0';gc;}x*=k;
}

const int N=100005;
const int M=500005;

vector<int> G[N];

int ac[N][20],dep[N],in[N],ptn[N],out[N],dfs_clock;
void dfs(int x,int f){
    ac[x][0]=f;
    dep[x]=dep[f]+1;
    in[x]=++dfs_clock;
    ptn[dfs_clock]=x;
    for(int i=1;(ac[x][i]=ac[ac[x][i-1]][i-1]);++i);
    for(int i=0;i<G[x].size();++i){
        int v=G[x][i];
        if(v!=f)dfs(v,x);
    }
    out[x]=dfs_clock;
}

inline int find(int x,int t){
    for(int i=16;~i;--i){
        if(dep[x]-dep[t]>(1<<i))x=ac[x][i];
    }
//  assert(ac[x][0]==t);
    return x;
}

int be[N],a[N],b[N],cntl[N],cntr[N];

struct Query{
    int l,r,type,id;
    bool operator < (Query x) const{
        return be[l]==be[x.l]?r<x.r:l<x.l;
    }
};

ll Ans[M];
vector<Query> Q;

int n,m,root=1;

inline void get_query(int x,vector<pair<int,int> > &ret){
    if(x==root)ret.push_back(make_pair(1,n));
    else if(in[x]<in[root]&&out[root]<=out[x]){
        int t=find(root,x);
        ret.push_back(make_pair(1,in[t]-1));
        ret.push_back(make_pair(out[t]+1,n));
    }
    else ret.push_back(make_pair(in[x],out[x]));
}

inline void query(int x,int y,int id){
    vector<pair<int,int> >T1,T2;
    get_query(x,T1);
    get_query(y,T2);
    for(int i=0;i<T1.size();++i){
        for(int j=0;j<T2.size();++j){
            Q.push_back(Query{T1[i].second,T2[j].second,1,id});
            if(T1[i].first>1)Q.push_back(Query{T1[i].first-1,T2[j].second,-1,id});
            if(T2[j].first>1)Q.push_back(Query{T1[i].second,T2[j].first-1,-1,id});
            if((T1[i].first>1)&&(T2[j].first>1))Q.push_back(Query{T1[i].first-1,T2[j].first-1,1,id});
        }
    }
}

int main(){
//  freopen(".in","r",stdin);
//  freopen(".out","w",stdout);
    r(n),r(m);
    int block_size=sqrt(n);
    for(int i=1;i<=n;++i){
        r(a[i]);
        b[i]=a[i];
        be[i]=(i-1)/block_size+1;
    }
    sort(b+1,b+n+1);
    int tot=unique(b+1,b+n+1)-b-1;
    for(int i=1;i<=n;++i)a[i]=lower_bound(b+1,b+tot+1,a[i])-b;
    for(int i=1,u,v;i<n;++i){
        r(u),r(v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs(1,0);
    int id=0;
    for(int i=1,opt,x,y;i<=m;++i){
        r(opt);
        if(opt==1)r(root);
        else {
            r(x),r(y);
            query(x,y,++id);
        }
    }
    
    sort(Q.begin(),Q.end());
     
    ll ans=0;
    int l=0,r=0;
    for(int i=0;i<Q.size();++i){
        Query &x=Q[i];
        
        while(l<x.l)cntl[a[ptn[++l]]]++,ans+=cntr[a[ptn[l]]];
        while(l>x.l)cntl[a[ptn[l]]]--,ans-=cntr[a[ptn[l--]]];
        while(r<x.r)cntr[a[ptn[++r]]]++,ans+=cntl[a[ptn[r]]];
        while(r>x.r)cntr[a[ptn[r]]]--,ans-=cntl[a[ptn[r--]]];
        
        Ans[x.id]+=x.type*ans;
    }
    for(int i=1;i<=id;++i)printf("%lld\n",Ans[i]);
}