首页 > 代码库 > bzoj4455【ZJOI2016】小星星

bzoj4455【ZJOI2016】小星星

题意:http://www.lydsy.com/JudgeOnline/problem.php?id=4455

   给一张图和该图的一棵生成树,求可能的编号方案数

sol  :dalao教导我们,看到计数想容斥

   考虑对点i标号,O(2^n)枚举可以标的顶点,然后进行树形dp

   f[i][j]表示点i标j号的方案数,容斥一发即可

   最后答案为∑f[1][i],记得开longlong

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
using namespace std;
const int Mx=410;
int n,m,cnt,a[Mx],map[Mx][Mx];
long long ans,f[Mx][Mx];
int tot,head[Mx],nxt[Mx],ver[Mx];
void add(int x,int y)
{
    nxt[++tot]=head[x];
    ver[tot]=y;
    head[x]=tot;
}
void dp(int x,int fa)
{
    for(int i=head[x];i;i=nxt[i])
    {  
        int y=ver[i];  
        if(y!=fa) dp(y,x);
    }  
    for(int i=1;i<=cnt;i++)
    {  
        f[x][i]=1;  
        for(int j=head[x];j;j=nxt[j])
        {
            int y=ver[j]; long long t=0;
            if(y!=fa)
            {
                for(int k=1;k<=cnt;k++)
                    if(map[a[i]][a[k]])
                        t+=f[y][k];
                f[x][i]*=t;
            }  
        }  
    }  
}  
int main()
{
    scanf("%d%d",&n,&m);  
    for(int i=1,x,y;i<=m;i++)
        scanf("%d%d",&x,&y),
        map[x][y]=map[y][x]=1;
    for(int i=1,x,y;i<n;i++)
        scanf("%d%d",&x,&y), 
        add(x,y),add(y,x);  
    for(int i=1;i<(1<<n);i++)
    {  
        cnt=0;long long t=0;  
        for(int j=1;j<=n;j++) 
            if((i>>(j-1))&1) a[++cnt]=j;
        dp(1,0);
        for(int j=1;j<=cnt;j++) t+=f[1][j];  
        ans+=t*(long long) (pow(-1,n-cnt));  
    }  
    printf("%lld\n",ans);
    return 0;  
}

 

bzoj4455【ZJOI2016】小星星