题目链接:BZOJ 3162
题目大意:求一棵无根树上本质不同的独立集个数
题解:经典的求树的独立集个数可以DP来做,f[x][0/1]分别表示以x为根的子树中不选点x和选点x的独立集个数,初始f[x][0]=f[x][1]=1
,转移的时候f[x][0]*=f[v][0]+f[v][1],f[x][1]*=f[v][0] /*v是x的子节点*/
。
这个题只需要考虑一下同构就可以了。判断树是否同构可以哈希。转移时若一个节点x有v_1,v_2,…v_m m个同构的子树,则f[x][1]*=C(m+f[v1][0]-1,m)
,即相当于f[v1][0]种物品,每种有无限个,从中取m个物品的本质不同方案数。同理f[x][0]*=C(m+f[v1][0]+f[v1][1]-1,m)
。为了方便可以选择无根树的重心作为根来DP(如果有两个重心就建一个虚拟节点作为根,两个重心连向它),以保证在任意一个点x,子树外的部分与x的各子树不会同构。
code(有参考hzwer大神(⊙_⊙))
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 500005
#define inf 1000000000
#define mod 1000000007
#define seed 9875321
using namespace std;
inline int read()
{
char c=getchar(); int num=0,f=1;
while (c<'0'||c>'9') { if (c=='-') f=-1; c=getchar(); }
while (c<='9'&&c>='0') { num=num*10+c-'0'; c=getchar(); }
return num*f;
}
struct edge{
int to,ne;
}e[N<<1];
int n,root,tot=1,head[N],h[2],siz[N],sta[N],top,mn=inf;
long long inv[N],f[N][2],ans;
unsigned long long H[N];
inline bool cmp(int x,int y) { return H[x]>H[y]; }
inline void push(int x,int y) { e[++tot].to=y; e[tot].ne=head[x]; head[x]=tot; }
void pre()
{
inv[1]=1;
for (int i=2;i<=n;i++)
inv[i]=(long long)inv[mod%i]*(mod-mod/i)%mod;
}
int C(int n,int m)
{
long long ret=1; n%=mod;
for (int i=1;i<=m;i++) ret=ret*(n-i+1)%mod*inv[i]%mod;
return ret;
}
inline int color(int n,long long m) { return C(n+m-1,n); }
void dfs(int now,int pre)
{
siz[now]=1; int tmp=0;
for (int i=head[now];i;i=e[i].ne)
{
int v=e[i].to; if (v==pre) continue;
dfs(v,now); siz[now]+=siz[v];
tmp=max(tmp,siz[v]);
}
tmp=max(tmp,n-siz[now]);
if (tmp<mn) mn=tmp,h[0]=now,h[1]=0;
else if (tmp==mn) h[1]=now;
}
void dp(int now,int pre)
{
f[now][0]=f[now][1]=1;
for (int i=head[now];i;i=e[i].ne)
{
int v=e[i].to; if (v==pre) continue;
dp(v,now);
}
int j; top=0;
for (int i=head[now];i;i=e[i].ne)
{
int v=e[i].to; if (v==pre) continue;
sta[++top]=v;
}
sort(sta+1,sta+top+1,cmp);
for (int i=1;i<=top;i=j)
{
for (j=i+1;j<=top&&H[sta[i]]==H[sta[j]];j++);
f[now][0]=f[now][0]*color(j-i,f[sta[i]][0]+f[sta[i]][1])%mod;
f[now][1]=f[now][1]*color(j-i,f[sta[i]][0])%mod;
}
H[now]=233;
for (int i=1;i<=top;i++) (((H[now]*=seed)+=H[sta[i]])^=H[sta[i]])+=H[sta[i]];
/*树哈希*/
}
int main()
{
n=read(); pre();
for (int i=1;i<n;i++)
{
int x=read(),y=read();
push(x,y); push(y,x);
}
dfs(1,0);
if (h[1])
{
for (int i=head[h[0]];i;i=e[i].ne)
if (e[i].to==h[1]) e[i].to=e[i^1].to=root=n+1;
push(n+1,h[0]); push(n+1,h[1]);
}
else root=h[0];
dp(root,0);
if (!h[1]) ans=(f[root][0]+f[root][1])%mod;
else
{
int x=h[0],y=h[1];
if (H[x]==H[y]) ans=(f[x][0]*f[y][1]%mod+color(2,f[x][0]))%mod;
else ans=(f[x][0]*f[y][0]%mod+f[x][1]*f[y][0]%mod+f[x][0]*f[y][1]%mod)%mod;
}
printf("%lld",ans);
return 0;
}