一、题目
二、解法
这道题不要被题目特点限制思路了,不一定是虚树的题目。
有一个简单的
d
p
dp
dp,设
d
p
[
i
]
[
j
]
dp[i][j]
dp[i][j]为前
i
i
i个数划分成
j
j
j组的方案数,转移考虑第
i
i
i个数怎么划分,他不能和他的祖先划分到一起,设他的祖先个数为
f
f
f,那么他可以划分到的组别有
m
a
x
(
0
,
j
−
f
)
max(0,j-f)
max(0,j−f)组(因为一个点的祖先和祖先也存在冲突,所以祖先和祖先一定在不同组别),还可以独自划分一组:
d
p
[
i
]
[
j
]
=
d
p
[
i
−
1
]
[
j
]
×
max
(
0
,
j
−
f
)
+
d
p
[
i
−
1
]
[
j
−
1
]
dp[i][j]=dp[i-1][j]\times\max(0,j-f)+dp[i-1][j-1]
dp[i][j]=dp[i−1][j]×max(0,j−f)+dp[i−1][j−1]这个
d
p
dp
dp是可以暴力跑的,问题的关键在于求
f
f
f,由于根是在变的,把每个点标记以后求这个点到根的路径上有多少标记点就行了。还要注意
d
p
dp
dp的顺序,一个点的祖先一定要比一个点先
d
p
dp
dp,需要排序,一个点祖先的祖先个数一定要比一个点的祖先个数少,所以按祖先个数排序即可(不能排
d
f
s
dfs
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);
}
}