P3349 [ZJOI2016]小星星

(Updated: )
oi
,

题目链接：P3349

给定一棵 $n$ 个节点的树和一个 $n$ 个节点的图，要求为树上的每个节点映射到一个图上的节点（双射），且要求树上边对应图上也必须有相应的边，求方案数。

$n\leq 17$。

Sol

考虑朴素 DP，设 $F[i][j][k]$​ 表示将 $i$ 映射到 $j$，且 $i$ 的子树内的映射集合为 $k$ 的方案数。

瓶颈在于枚举子集，时间复杂度为 $\mathcal O(n^3\times 3^n)$。

考虑去掉 $k$ 这一维，显然直接套用容斥原理即可。

总方案数即为 全集 $-$ 少选一个 $+$ 少选两个 $-\dots$

这样时间复杂度降到了 $\mathcal O(n^3\times 2^n)$。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#include<bits/stdc++.h>
#define Tp template<typename Ty>
#define Ts template<typename Ty,typename... Ar>
#define W while
#define I inline
#define RI register int
#define LL long long
#define Cn const
#define CI Cn int&
#define gc getchar
#define D isdigit(c=gc())
#define pc(c) putchar((c))
#define min(x,y) ((x)<(y)?(x):(y))
#define max(x,y) ((x)>(y)?(x):(y))
using namespace std;
namespace Debug{
    Tp I void _debug(Cn char* f,Ty t){cerr<<f<<'='<<t<<endl;}
    Ts I void _debug(Cn char* f,Ty x,Ar... y){W(*f!=',') cerr<<*f++;cerr<<'='<<x<<",";_debug(f+1,y...);}
    Tp ostream& operator<<(ostream& os,Cn vector<Ty>& V){os<<"[";for(Cn auto& vv:V) os<<vv<<",";os<<"]";return os;}
    #define gdb(...) _debug(#__VA_ARGS__,__VA_ARGS__)
}using namespace Debug;
namespace FastIO{
    Tp I void read(Ty& x){char c;int f=1;x=0;W(!D) f=c^'-'?1:-1;W(x=(x<<3)+(x<<1)+(c&15),D);x*=f;}
    Ts I void read(Ty& x,Ar&... y){read(x),read(y...);}
    Tp I void write(Ty x){x<0&&(pc('-'),x=-x,0),x<10?(pc(x+'0'),0):(write(x/10),pc(x%10+'0'),0);}
    Tp I void writeln(Cn Ty& x){write(x),pc('\n');}
}using namespace FastIO;
Cn int N=18;
int n,m,H[N][N],A[N];
vector<int> G[N],V;
LL Ans,F[N][N];
I int Count(RI x){RI X=0;W(x) x-=(x&-x),X++;return X;}
I void Dfs(CI x,CI fa){
    LL X;for(auto i:G[x]) i^fa&&(Dfs(i,x),0);
    for(auto i:V){F[x][i]=1;for(auto j:G[x]) if(j^fa){X=0;for(auto k:V) if(i^k&&H[i][k]) X+=F[j][k];F[x][i]*=X;}}
}
I LL S(CI x){
    RI i;LL X=0;for(V.clear(),memset(A,0,sizeof(A)),i=1;i<=n;i++) x>>(i-1)&1&&(V.push_back(i),A[i]=1);
    Dfs(1,0);for(auto i:V) X+=F[1][i];return X;
}
int main(){
    RI i,x,y;for(read(n,m),i=1;i<=m;i++) read(x,y),H[x][y]=H[y][x]=1;
    for(i=1;i<n;i++) read(x,y),G[x].push_back(y),G[y].push_back(x);
    for(i=0;i<(1<<n);i++) Ans+=((n-Count(i))&1?-S(i):S(i));return writeln(Ans),0;
}