BZOJ 4455: [Zjoi2016]小星星 容斥原理+樹形DP

題意： 

給定一張圖，和這個圖的一顆生成樹.  

求：從新對這顆樹標號，有多少種標號方式知足若是一條邊 $(x,y)$ 在樹中出現的話在圖中也出現.         

暴力作法：$f[i][j][S]$ 即在以 1 爲根的樹中遍歷到點 $i$，該點映射的點是 $j$，且 $i$ 這顆子樹映射的點集是 $S$.       

轉移的話時間複雜度比較高，並且很是麻煩.       

咱們之全部有 $S$ 的限制是由於不能夠出現樹上兩個不一樣的點同時映射了相同的點.   

若是不考慮映射相同的點的話直接設狀態 $f[i][j]$ 便可.    

$f[i][j]=\prod_{k \subseteq son[i]} f[k][a]$，$E(j,a)$.               

不妨考慮容斥：令 $f(k)$ 表示用了最多 $k$ 個點的映射數量，那麼 $Ans=\sum_{i=1}^{n} (-1)^{n-i} f(i).$     

求 $f(k)$ 的話就枚舉 $n$ 的子集，而後每次只容許 $f[i][j]$ 中的 $j$ 是子集中的元素.        

#include <cstdio>  
#include <cstring>
#include <algorithm> 
#define N 18   
#define ll long long 
#define setIO(s) freopen(s".in","r",stdin)    
using namespace std;   
int n,m,edges;  
int del[N],con[N][N],hd[N],to[N<<1],nex[N<<1]; 
ll dp[N][N];     
void add(int u,int v) 
{
    nex[++edges]=hd[u],hd[u]=edges,to[edges]=v;  
}
int count(int x) 
{             
    memset(del,0,sizeof(del));  
    int cn=0,re=1,sz=0; 
    for(;x;x>>=1,++re)  if(x&1) del[re]=1,++sz;    
    return sz;   
}      
void dfs(int x,int ff) 
{ 
    for(int i=1;i<=n;++i) if(!del[i]) dp[x][i]=1;   
    for(int i=hd[x];i;i=nex[i]) 
    {
        int y=to[i]; 
        if(y==ff) continue;     
        dfs(y,x);     
        for(int j=1;j<=n;++j)
        {
            if(del[j]) continue; 
            ll sum=0;  
            for(int k=1;k<=n;++k) 
            {
                if(del[k]||!con[j][k]) continue;  
                sum+=dp[y][k];    
            }      
            dp[x][j]*=sum;  
        }
    }
}
int main() 
{      
    // setIO("input"); 
    int i,j; 
    scanf("%d%d",&n,&m); 
    for(i=1;i<=m;++i) 
    {
        int x,y;
        scanf("%d%d",&x,&y),con[x][y]=con[y][x]=1;  
    }  
    for(i=1;i<n;++i) 
    {
        int x,y; 
        scanf("%d%d",&x,&y);   
        add(x,y),add(y,x);   
    }
    ll ans=0;  
    for(i=0;i<(1<<n);++i) 
    {
        int id=count(i);           
        dfs(1,0);  
        ll sum=0ll;   
        for(j=1;j<=n;++j) if(!del[j]) sum+=dp[1][j];     
        if(id&1) ans-=sum; 
        else ans+=sum;   
    }   
    printf("%lld\n",ans); 
    return 0;   
}