[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的结合版
你可以把一个询问拆成九个搞
然后卡了半小时常数。。
#includestruct 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]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]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;iregister 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(lwhile(l>A[i].l)dell(l--);
while(rwhile(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;
}