BZOJ 3162 独钓寒江雪(树形DP)

题目链接: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;
}
阅读更多
版权声明:转载请标明出处,O(∩_∩)O谢谢~ https://blog.csdn.net/XHRlyb/article/details/79956781
文章标签: dp 哈希
个人分类: 树形dp DP
上一篇BZOJ 3743 Kamp(树形DP)
下一篇BZOJ 3197 assassin(树形DP+费用流)
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭