CF1111E Tree

一、题目

点此看题

二、解法

这道题不要被题目特点限制思路了，不一定是虚树的题目。

有一个简单的$dp$，设$dp[i][j]$为前$i$个数划分成$j$组的方案数，转移考虑第$i$个数怎么划分，他不能和他的祖先划分到一起，设他的祖先个数为$f$，那么他可以划分到的组别有$max(0,j-f)$组（因为一个点的祖先和祖先也存在冲突，所以祖先和祖先一定在不同组别），还可以独自划分一组：
$$dp[i][j]=dp[i-1][j]\times \max (0,j-f)+dp[i-1][j-1]$$这个$dp$是可以暴力跑的，问题的关键在于求$f$，由于根是在变的，把每个点标记以后求这个点到根的路径上有多少标记点就行了。还要注意$dp$的顺序，一个点的祖先一定要比一个点先$dp$，需要排序，一个点祖先的祖先个数一定要比一个点的祖先个数少，所以按祖先个数排序即可（不能排$dfs$序，因为根在变）

#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
const int M = 100005;
const int MOD = 1e9+7;
#define int long long
int read()
{
	int x=0,f=1;char c;
	while((c=getchar())<'0' || c>'9') {if(c=='-') f=-1;}
	while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
	return x*f;
}
int n,q,tot,Index,f[M],a[M],fa[M],siz[M];
int F[M],dep[M],son[M],top[M],num[M],sum[4*M],dp[M][305];
struct edge
{
	int v,next;
}e[2*M];
bool cmp(int a,int b)
{
	return F[a]<F[b];
}
void dfs1(int u,int p)
{
	fa[u]=p;siz[u]=1;
	dep[u]=dep[p]+1;
	for(int i=f[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(v==p) continue;
		dfs1(v,u);
		siz[u]+=siz[v];
		if(siz[son[u]]<siz[v])
			son[u]=v;
	}
}
void dfs2(int u,int tp)
{
	top[u]=tp;
	num[u]=++Index;
	if(son[u]) dfs2(son[u],tp);
	for(int i=f[u];i;i=e[i].next)
		if(e[i].v^fa[u] && e[i].v^son[u])
			dfs2(e[i].v,e[i].v);
}
void add(int i,int l,int r,int id,int v)
{
	if(l==r)
	{
		sum[i]+=v;
		return ;
	}
	int mid=(l+r)>>1;
	if(mid>=id) add(i<<1,l,mid,id,v);
	else add(i<<1|1,mid+1,r,id,v);
	sum[i]=sum[i<<1]+sum[i<<1|1];
}
int query(int i,int l,int r,int L,int R)
{
	if(L>r || l>R) return 0;
	if(L<=l && r<=R) return sum[i];
	int mid=(l+r)>>1;
	return query(i<<1,l,mid,L,R)+query(i<<1|1,mid+1,r,L,R);
}
int ask(int u,int v)
{
	int ans=0;
	while(top[u]^top[v])
	{
		if(dep[top[u]]<=dep[top[v]]) swap(u,v);
		ans+=query(1,1,n,num[top[u]],num[u]);
		u=fa[top[u]];
	}
	if(dep[u]>=dep[v]) swap(u,v);
	ans+=query(1,1,n,num[u],num[v]);
	return ans;
}
signed main()
{
	n=read();q=read();
	for(int i=1;i<n;i++)
	{
		int u=read(),v=read();
		e[++tot]=edge{v,f[u]},f[u]=tot;
		e[++tot]=edge{u,f[v]},f[v]=tot;
	}
	dfs1(1,0);
	dfs2(1,1);
	while(q--)
	{
		int k=read(),m=read(),r=read();
		for(int i=1;i<=k;i++)
		{
			a[i]=read();
			add(1,1,n,num[a[i]],1);
		}
		for(int i=1;i<=k;i++)
			F[a[i]]=ask(a[i],r)-1;
		sort(a+1,a+1+k,cmp);
		for(int i=1;i<=k;i++)
			for(int j=1;j<=m;j++)
				dp[i][j]=0;
		int ans=0;dp[1][1]=1;
		for(int i=2;i<=k;i++)
		{
			int tmp=F[a[i]];
			for(int j=1;j<=min(i,m);j++)
				dp[i][j]=(1ll*max(0ll,j-tmp)*dp[i-1][j]%MOD+dp[i-1][j-1])%MOD;
		}
		for(int i=1;i<=k;i++)
			add(1,1,n,num[a[i]],-1);
		for(int i=1;i<=m;i++)
			ans=(ans+dp[k][i])%MOD;
		printf("%lld\n",ans);
	}
}