正题
题目大意
nnn个点的一棵树,定义f(S)f(S)f(S)表示点集SSS的生成子图中的边数量。
求∑S∈Vf(S)k\sum_{S\in V}f(S)^kS∈V∑f(S)k
解题思路
因为kkk很小,所以可以考虑一下二项式拆解,我们需要快速的计算出(a+b)k(a+b)^k(a+b)k,那么就需要求出ai,bi(i≤k)a^i,b^i(i\leq k)ai,bi(i≤k)。
那么我们可以设f0/1,i,jf_{0/1,i,j}f0/1,i,j表示点iii是否被选择时f(S)jf(S)^jf(S)j的答案。
这样我们可以做到O(k2)O(k^2)O(k2)从yyy转移到xxx。具体的,对于f0,x,if_{0,x,i}f0,x,i,我们把它和f0,y,i+f1,y,if_{0,y,i}+f_{1,y,i}f0,y,i+f1,y,i卷起来。对于f1,x,if_{1,x,i}f1,x,i,我们先把f1,y,if_{1,y,i}f1,y,i和111卷起来,然后同上。
时间复杂度O(nk2)O(nk^2)O(nk2)
codecodecode
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e5+10,XJQ=998244353;
struct node{int to,next;
}a[N*2];
int n,k,tot,m,ls[N],f[2][N][12],c[11][12],tmp[4][12];
void addl(int x,int y){a[++tot].to=y;a[tot].next=ls[x];ls[x]=tot;return;
}
void dp(int x,int fa){f[0][x][0]=f[1][x][0]=1;for(int p=ls[x];p;p=a[p].next){int y=a[p].to;if(y==fa)continue;dp(y,x);for(int i=0;i<=k;i++)tmp[2][i]=f[1][y][i];for(int i=0;i<=k;i++){tmp[3][i]=0;for(int j=0;j<=i;j++)(tmp[3][i]+=1ll*tmp[2][j]%XJQ*c[i][j]%XJQ)%=XJQ;}for(int i=0;i<=k;i++)tmp[0][i]=f[0][x][i],tmp[1][i]=f[1][x][i];for(int i=0;i<=k;i++){f[0][x][i]=f[1][x][i]=0;for(int j=0;j<=i;j++){(f[0][x][i]+=1ll*tmp[0][i-j]*(f[0][y][j]+f[1][y][j])%XJQ*c[i][j]%XJQ)%=XJQ;(f[1][x][i]+=1ll*tmp[1][i-j]*(tmp[3][j]+f[0][y][j])%XJQ*c[i][j]%XJQ)%=XJQ;}}}return;
}
int main()
{freopen("subgraph.in","r",stdin);
// freopen("subgraph.out","w",stdout);scanf("%d%d%d",&n,&m,&k);for(int i=1;i<=m;i++){int x,y;scanf("%d%d",&x,&y);addl(x,y);addl(y,x);}c[0][0]=1;for(int i=1;i<=k;i++)for(int j=0;j<=i;j++){c[i][j]=c[i-1][j];if(j)(c[i][j]+=c[i-1][j-1])%=XJQ;}dp(1,0);printf("%d\n",(f[0][1][k]+f[1][1][k])%XJQ);return 0;
}