【題解】Luogu P3349 [ZJOI2016]小星星

原題傳送門

咱們考慮設\(dp_{i,j}\)表示樹上的點\(i\)在圖上對應的點爲\(j\)時\(i\)和子樹對應在圖上的方案數

\(dp_{u_i}=\prod_{v \in u.son} dp_{v,j}*w[i][j]\)（w[i][j]表示在圖中\(i\)和\(j\)是否連通）

這個dp的複雜度是\(O(n^3)\)，可是會發現它會多算答案，由於這個算法有可能會將多個樹上的點對應到一個圖上的點

咱們珂以二進制枚舉，枚舉哪些圖上的點參與dp，按照容斥原理加減，這樣就珂以算出答案，複雜度爲\(O(2^n n^3)\)

#include <bits/stdc++.h>
#define ll long long
#define N 20
#define getchar nc
using namespace std;
inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
    register int x=0,f=1;register char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return x*f;
}
inline void write(register ll x)
{
    if(!x)putchar('0');if(x<0)x=-x,putchar('-');
    static int sta[20];register int tot=0;
    while(x)sta[tot++]=x%10,x/=10;
    while(tot)putchar(sta[--tot]+48);
}
struct edge{
    int to,next;
}e[N<<1];
int head[N],cnt=0;
inline void add(register int u,register int v)
{
    e[++cnt]=(edge){v,head[u]};
    head[u]=cnt;
}
int n,m,w[N][N],ban[N],vis[N];
ll f[N][N],ans;
inline void dfs(register int x)
{
    vis[x]=1;
    for(register int i=1;i<=n;++i)
        f[x][i]=1;
    for(register int i=head[x];i;i=e[i].next)
    {
        int v=e[i].to;
        if(vis[v])
            continue;
        dfs(v);
        for(register int j=1;j<=n;++j)
        {
            ll sum=0;
            for(register int k=1;k<=n;++k)
                sum+=f[v][k]*(w[k][j]&ban[k]&ban[j]);
            f[x][j]*=sum;
        }
    }
}
int main()
{
    n=read(),m=read();
    for(register int i=1;i<=m;++i)
    {
        int u=read(),v=read();
        w[u][v]=w[v][u]=1;
    }
    for(register int i=1;i<n;++i)
    {
        int u=read(),v=read();
        add(u,v),add(v,u);
    }
    for(register int k=1,siz;k<=(1<<n)-1;++k)
    {
        siz=n;
        for(register int i=1;i<=n;++i)
            ban[i]=0;
        for(register int i=1,p=k;p;p>>=1,++i)
            ban[i]=p&1,siz-=p&1;
        for(register int i=1;i<=n;++i)
            vis[i]=0;
        dfs(1);
        ll cnt=0;
        for(register int i=1;i<=n;++i)
            cnt+=f[1][i];
        if(siz%2)
            ans-=cnt;
        else
            ans+=cnt;
    }
    write(ans);
    return 0;
}