Codeforces Round #721 (Div. 2) D. MEX Tree(思维题-树、mex性质)

题目

t(t<=1e4)组样例，每次给出n(n<=2e5)个点，点号[0,n-1]的一棵树，

对于每一个k从[0,n+1]，求树上路径满足路径点集合的mex恰为k的无序点对数

保证sumn<=2e5

思路来源

乱搞AC

题解

如果mex>=1，则这条路径一定要包含0

如果mex>=2，则这条路径一定要包含0、1

如果mex>=3，则这条路径一定要包含0、1、2

则注意到mex=i能成立，必须[0,i-1]在一条链上，

所以对0号点dfs，时刻维护以0为根的两条链的一端，

fi表示第一条链的尾端，se表示第二条链的尾端，

每次插入时判断，是位于链中，还是成为了新的链尾，这里用lca判断

如果出现了三条链就说明此时的mex值不存在了，break即可

求出mex>=i的之后，用mex>=i减去mex>=i+1差分一下即可得mex=i的

mex>=1的是单独求的，即位于0两侧的点对数，其中一侧可以包含0

mex>=2开始，就维护0 1所在的一条链了，

代码

#include<bits/stdc++.h>
using namespace std;
#define pb push_back
typedef long long ll;
const int N=2e5+10,lg=19;
int t,n,u,v,sz[N],dep[N],f[N][lg],son;
vector<int>e[N];
int fi,se;//时刻维护当前mex是不是一条链 两个儿子点分别是fi se 
ll suf[N];//suf[i]:mex>=i的方案数 即必过[0,i-1] 
void dfs(int u,int fa){
	sz[u]=1;
	f[u][0]=fa;
	for(int i=1;i<lg;++i){
		if(f[u][i-1]==-1)f[u][i]=-1;
		else f[u][i]=f[f[u][i-1]][i-1];
	}
	for(auto &v:e[u]){
		if(v==fa)continue;
		dep[v]=dep[u]+1;
		dfs(v,u);
		if(u==0){
			suf[1]+=1ll*sz[u]*sz[v];
		}
		sz[u]+=sz[v];
	}
}

int lca(int x,int y){
	if(dep[x]<dep[y])swap(x,y);
	int d=dep[x]-dep[y];
	for(int i=lg-1;i>=0;--i){
		if(d>>i&1)x=f[x][i];
	}
	if(x==y)return x;
	for(int i=lg-1;i>=0;--i){
		if(f[x][i]!=f[y][i]){
			x=f[x][i],y=f[y][i];
		}
	}
	return f[x][0];
}
int main(){
	scanf("%d",&t);
	while(t--){
		scanf("%d",&n);
		suf[0]=1ll*n*(n-1)/2;
		fi=se=son=-1;//链的左右端点 
		for(int i=0;i<=n;++i){
			e[i].clear();
			suf[i]=dep[i]=0;
		}
		for(int i=1;i<n;++i){
			scanf("%d%d",&u,&v);
			e[u].pb(v);e[v].pb(u);
		}
		dfs(0,-1);
		suf[0]=1ll*n*(n-1)/2-suf[1];
		for(int i=1;i<n;++i){
			if(fi==-1){
				fi=i;
				//son为i到0的这条链上 离i最近的点 
				son=i;
				for(int j=lg-1;j>=0;--j){
					if(f[son][j]==-1)continue;
					if(f[son][j]){
						son=f[son][j];
					}
				}
				//printf("son:%d\n",son); 
				suf[i+1]=1ll*(n-sz[son])*sz[fi];
			}
			else if(se==-1){
				int anc=lca(fi,i);
				if(anc==i){
					suf[i+1]=1ll*(n-sz[son])*sz[fi];
				}
				else if(anc==fi){
					fi=i;
					suf[i+1]=1ll*(n-sz[son])*sz[fi];
				}
				else if(anc==0){
					se=i;
					suf[i+1]=1ll*sz[fi]*sz[se];
				}
				else{
					break;
				}
			}
			else{
				int anc=lca(fi,i),anc2=lca(se,i);
				if(anc==i || anc2==i){
					suf[i+1]=1ll*sz[fi]*sz[se];
				}
				else if(anc==fi){
					fi=i;
					suf[i+1]=1ll*sz[fi]*sz[se];
				}
				else if(anc2==se){
					se=i;
					suf[i+1]=1ll*sz[fi]*sz[se];
				}
				else{
					break;
				}
			}
		}
		for(int i=1;i<n;++i){
			suf[i]-=suf[i+1];
		}
		for(int i=0;i<=n;++i){
			printf("%lld%c",suf[i]," \n"[i==n]);
		}
	}
	return 0;
}