BZOJ 4455 & UOJ 185 [ZJOI2016]小星星

容斥原理+树形DP

首先有一个暴力的想法，f[i][j][sta]表示树中i与图中k对应，i的子树与sta(状压)对应，然后大力转移。这需要枚举子集，复杂度约$O(n^2*3^n)$

考虑枚举状态sta表示图中可用点范围,记f[i][j]表示i和j对应，但对应可以重复，即可以不一一对应。

最终答案就是与全集一一对应的方案数。

考虑容斥，记$S_n$表示不使用图中节点n的方案，显然这些都是不合理的方案，我们只需要把总方案减去这些方案的并集即可。并集？容斥搞。

#include<cstdio>
#define N 20
using namespace std;
struct edge{int next, to;}e[N<<1];
int ecnt=1, last[N], pre[N], cnt=0, bin[N], n, m;
bool d[N][N];
long long f[N][N], ans=0;
void add(int a, int b)
{
    e[++ecnt]=(edge){last[a],b};
    last[a]=ecnt;
}
void dp(int x, int fa)
{
    for(int i = last[x]; i; i=e[i].next)
        if(e[i].to!=fa)
            dp(e[i].to,x);
    for(int i = 1; i <= cnt; i++)
    {
        f[x][i]=1;
        for(int p = last[x]; p; p=e[p].next)
        {
            long long sum=0;
            int y = e[p].to;
            if(y==fa)continue;
            for(int j = 1; j <= cnt; j++)
                if(d[pre[i]][pre[j]])sum+=f[y][j];
            f[x][i]*=sum;
        }
    }
}
int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= m; i++)
    {
        int a, b;
        scanf("%d%d",&a,&b);
        d[a][b]=d[b][a]=1;
    }
    for(int i = 1; i < n; i++)
    {
        int a, b;
        scanf("%d%d",&a,&b);
        add(a,b);
        add(b,a);
    }
    bin[0]=1;
    for(int i = 1; i < N; i++)
        bin[i]=bin[i-1]<<1;
    for(int s = 1, ss = 1<<n; s < ss; s++)
    {
        cnt=0;
        for(int i = 1; i <= n; i++)
            if(s&bin[i-1])
                pre[++cnt]=i;
        dp(1,-1);
        if(1&(cnt^n))
            for(int i = 1; i <= cnt; i++)
                ans-=f[1][i];
        else
            for(int i = 1; i <= cnt; i++)
                ans+=f[1][i];
    }
    printf("%lld\n",ans);
}