[BZOJ4455][ZJOI2016]小星星 容斥原理+树型DP

26 篇文章 0 订阅
8 篇文章 0 订阅

先考虑一个错误的DP, fi,j 表示子树 i 中,i映射到 j 的方案数,这样可能一个点被重复映射。
我们考虑多一维状态fi,j,S表示映射点集至多为 S 的方案,枚举S来DP,用朴素的容斥原理来求出正确答案即可。
代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long
using namespace std;
int n,m,st[20],top;
ll f[20][20],ans;
bool mp[20][20];
struct edge
{
    int t;
    edge *next;
}*con[20];
void ins(int x,int y)
{
    edge *p=new edge;
    p->t=y;
    p->next=con[x];
    con[x]=p;
}
void dp(int v,int fa,int s)
{
    for(int i=1;i<=top;i++)  f[v][st[i]]=1;
        for(edge *p=con[v];p;p=p->next)
            if(p->t!=fa)
            {
                dp(p->t,v,s);
                for(int i=1;i<=top;i++)
                {
                    ll cal=0;
                        for(int j=1;j<=top;j++)
                            cal+=f[p->t][st[j]]*mp[st[i]][st[j]];
                    f[v][st[i]]*=cal;
                }
            }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        mp[x][y]=mp[y][x]=1;
    }
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        ins(x,y);ins(y,x);
    }
    for(int s=0;s<(1<<n);s++)
    {
        top=0;
        for(int i=0;i<n;i++)
            if((s>>i)&1) st[++top]=i+1;
        dp(1,0,s);
        for(int i=1;i<=top;i++)
            ans+=f[1][st[i]]*((n-top)&1?-1:1);  
    }
    printf("%lld",ans);
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值