BZOJ3162 独钓寒江雪(哈希+树形dp)

  数独立集显然是可以树形dp的,问题在于本质不同。

  假设已经给树确立了一个根并且找到了所有等效(注意是等效而不是同构)子树,那么对转移稍加修改使用隔板法就行了。

  关键在于找等效子树。首先将树的重心(若有两个则加一个点作为唯一重心)作为根。这样任意极大等效子树(比如某两个等效子树里面的一部分等效,那么里面这一部分就不是极大的)一定有相同的父亲,否则我们所选的根是肯定存在一棵子树大小大于树的一半的,与重心性质矛盾。那么判等效就只需要考虑子树内同构了。

  同构判断采取哈希。这里使用最简单的类似字符串哈希的做法,用子树大小哈希。在保证同构树哈希值相同的前提下尽量增加变数。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
int read()
{
int x=,f=;char c=getchar();
while (c<''||c>'') {if (c=='-') f=-;c=getchar();}
while (c>=''&&c<='') x=(x<<)+(x<<)+(c^),c=getchar();
return x*f;
}
#define N 500010
#define P 1000000007
#define ul unsigned long long
int n,p[N],p_new[N],f[N][],size[N],q[N],inv[N],root,t=;
ul hash[N];
struct data{int to,nxt;
}edge[N<<],edge_new[N<<];
void addedge(int x,int y){t++;edge[t].to=y,edge[t].nxt=p[x],p[x]=t;}
void addedge_new(int x,int y){t++;edge_new[t].to=y,edge_new[t].nxt=p_new[x],p_new[x]=t;}
void dfs(int k,int from)
{
size[k]=;
for (int i=p[k];i;i=edge[i].nxt)
if (edge[i].to!=from)
{
dfs(edge[i].to,k);
size[k]+=size[edge[i].to];
}
}
void dfs2(int k,int from)
{
for (int i=p[k];i;i=edge[i].nxt)
if (edge[i].to!=from)
{
addedge_new(k,edge[i].to);
dfs2(edge[i].to,k);
}
}
int findroot(int k,int from)
{
int mx=;
for (int i=p[k];i;i=edge[i].nxt)
if (edge[i].to!=from&&size[edge[i].to]>size[mx]) mx=edge[i].to;
if ((size[mx]<<)<=n) return k;
else return findroot(mx,k);
}
int C(int n,int m)
{
int ans=;
for (int i=n;i>=n-m+;i--) ans=1ll*ans*i%P;
return 1ll*ans*inv[m]%P;
}
bool cmp(const int&a,const int&b)
{
return hash[a]<hash[b];
}
void gethash(int k)
{
int cnt=;
for (int i=p[k];i;i=edge[i].nxt)
q[++cnt]=hash[edge[i].to];
sort(q+,q+cnt+);
for (int i=;i<=cnt;i++) hash[k]=hash[k]*+q[i];
hash[k]=(hash[k]*+size[k])%P;
}
void dp(int k)
{
for (int i=p[k];i;i=edge[i].nxt)
dp(edge[i].to),gethash(edge[i].to);
int cnt=;
for (int i=p[k];i;i=edge[i].nxt)
q[++cnt]=edge[i].to;
sort(q+,q+cnt+,cmp);
f[k][]=f[k][]=;
for (int i=;i<=cnt;i++)
{
int t=i;
while (t<cnt&&hash[q[t+]]==hash[q[i]]) t++;
f[k][]=1ll*f[k][]*C(f[q[i]][]+t-i,t-i+)%P;
f[k][]=1ll*f[k][]*C(f[q[i]][]+f[q[i]][]+t-i,t-i+)%P;
i=t;
}
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("bzoj3162.in","r",stdin);
freopen("bzoj3162.out","w",stdout);
const char LL[]="%I64d\n";
#else
const char LL[]="%lld\n";
#endif
n=read();
inv[]=inv[]=;for (int i=;i<=n;i++) inv[i]=P-1ll*(P/i)*inv[P%i]%P;
for (int i=;i<=n;i++) inv[i]=1ll*inv[i]*inv[i-]%P;
for (int i=;i<n;i++)
{
int x=read(),y=read();
addedge(x,y),addedge(y,x);
}
dfs(,);
root=findroot(,);
dfs(root,root);
int v=;
for (int i=p[root];i;i=edge[i].nxt)
if (!(n&)&&size[edge[i].to]==(n>>)) {v=edge[i].to;break;}
t=;
if (v)
{
n++;dfs2(root,v);dfs2(v,root);
addedge_new(n,root),addedge_new(n,v);root=n;
}
else dfs2(root,root);
memcpy(p,p_new,sizeof(p));
memcpy(edge,edge_new,sizeof(edge));
dfs(root,root);
dp(root);
if (v)
{
int x=edge[p[root]].to,y=edge[edge[p[root]].nxt].to;
if (hash[x]==hash[y])
f[root][]=(C(f[x][]+,)+1ll*f[x][]*f[x][]%P)%P;
else f[root][]=(1ll*f[x][]*(f[y][]+f[y][])%P+1ll*f[x][]*f[y][]%P)%P;
f[root][]=;
}
cout<<(f[root][]+f[root][])%P;
return ;
}
上一篇:HDU 5212 Code (莫比乌斯反演)


下一篇:[ACM_模拟] ACM - Draw Something Cheat [n个长12的大写字母串,找出交集,按字母序输出]