[ARC101E]Ribbons on Tree(容斥,dp)

Description

给定一棵有 \(n\) 个节点的树,满足 \(n\) 为偶数。初始时,每条边都为白色。

现在请你将这些点两两配对成 \(\frac{n}{2}\) 个无序点对。每个点对之间的的路径都会被染成黑色

求有多少种配对方案,使得树上没有白边?

\(n\le 5000\)

Solution

法一:

树上的路径很难直接考虑。

有一种容斥的做法:记边集为 E ,枚举 T 子集中的边强制为白边,其余的不作限制, 那么:

\[Ans = \sum_{T\subseteq E} (-1) ^ {T} F(T) \]

\(F(T)\) 为强制 T 的边为白边的方案数。

把 T 删掉后不难发现树变成了若干个联通块,显然这若干个连通块是独立的。

对于一个大小为 n 的连通块,两点随便配对的方案数是 \((n - 1) * (n - 3) * \cdots * 1\),记为 \(g(n)\)

然而暴力枚举 T 复杂度过高,考虑树型 dp ,需要知道的状态是 u 当前所在联通块大小以及容斥系数(即 T 的奇偶)。

\(dp[u][i][0/1]\) 为 u 子树内,u 所在联通块大小为 i ,T 的奇偶性是 0 / 1 的方案数。

转移就合并 u 的子树 v ,同时考虑 <u, v> 这条边是否选入 T 集合,有点做 01 背包的感觉。

\[dp[v][i][a]\times dp[u][i][b] \rightarrow dp'[u][i + j][a\oplus b]\\ dp[v][j][a]\times dp[u][i][b]\times g[i] \rightarrow dp'[u][i][a\oplus b\oplus 1] \]

最后答案就是 \(|T|\) 为偶数的 - \(|T|\) 为奇数的。

\[Ans = \sum_{i = 1} ^ n (dp[root][i][0] - dp[root][i][1]) \times g[i] \]

#include <cstdio>
#include <cstring>
#include <vector>
#include <iostream>
#include <algorithm>

using namespace std;

#define End exit(0)
#define LL long long
#define mp make_pair
#define SZ(x) ((int) x.size())
#define GO cerr << "GO" << endl
#define DE(x) cout << #x << " = " << x << endl
#define DEBUG(...) fprintf(stderr, __VA_ARGS__)

void proc_status()
{
	freopen("/proc/self/status","r",stdin);
	string s; while(getline(cin, s)) if (s[2] == 'P') { cerr << s << endl; return; }
}

template<typename T> inline T read() 
{
	register T x = 0;
	register char c; register int f(1);
	while (!isdigit(c = getchar())) if (c == '-') f = -1;
	while (x = (x << 1) + (x << 3) + (c ^ 48), isdigit(c = getchar()));
	return x * f;
}

template<typename T> inline bool chkmin(T &a,T b) { return a > b ? a = b, 1 : 0; }
template<typename T> inline bool chkmax(T &a,T b) { return a < b ? a = b, 1 : 0; }

const int maxN = 5000 + 2;
const int mod = 1e9 + 7;

vector<int> adj[maxN + 2];
int g[maxN + 2], n, size[maxN + 2];
int dp[maxN + 2][maxN + 2][2];

void input()
{
    n = read<int>();
    for (int i = 1; i < n; ++i)
    {
        int u = read<int>(), v = read<int>();
        adj[u].push_back(v), adj[v].push_back(u);
    }
}

void dfs(int u, int f)
{
    static int tmp[maxN + 2][2];

    size[u] = 1;
    dp[u][1][0] = 1;
    for (int v : adj[u])
        if (v != f)
        {
            dfs(v, u);
            for (int i = 0; i <= size[u]; ++i)
                for (int j = 0; j <= size[v]; ++j)
                    for (int a = 0; a < 2; ++a)
                        for (int b = 0; b < 2; ++b)
                        {
                            (tmp[i + j][a ^ b] += (LL) dp[v][j][a] * dp[u][i][b] % mod) %= mod;
                            if (!(j & 1))
                                (tmp[i][a ^ b ^ 1] += (LL) dp[v][j][a] * dp[u][i][b] % mod * g[j] % mod) %= mod;
                        }
            size[u] += size[v];
            for (int i = 0; i <= size[u]; ++i)
                for (int j = 0; j < 2; ++j)
                    dp[u][i][j] = tmp[i][j], tmp[i][j] = 0;
        }
}

void solve()
{
    g[0] = 1;
    for (int i = 2; i <= n; i += 2) g[i] = (LL) g[i - 2] * (i - 1) % mod;
    dfs(1, 0);
    int ans = 0;
    for (int i = 1; i <= n; ++i)
        (ans += ((LL) dp[1][i][0] - dp[1][i][1] + mod) * g[i] % mod) %= mod;
    cout << ans << endl;
}

int main() 
{
#ifndef ONLINE_JUDGE
	freopen("xhc2.in", "r", stdin);
	freopen("xhc2.out", "w", stdout);
#endif
    input();
    solve();
	return 0;
}

法二:

还是基于上面的容斥。

\(dp[u][i]\) 为 u 子树内还有 i 个点没有匹配,但考虑了容斥系数的答案。

合并子树后注意下 \(dp[u][0]\) 的转移要乘以 -1 的容斥系数(根除外)

#include <cstdio>
#include <cstring>
#include <vector>
#include <iostream>
#include <algorithm>

using namespace std;

#define End exit(0)
#define LL long long
#define mp make_pair
#define SZ(x) ((int) x.size())
#define GO cerr << "GO" << endl
#define DE(x) cout << #x << " = " << x << endl
#define DEBUG(...) fprintf(stderr, __VA_ARGS__)

void proc_status()
{
	freopen("/proc/self/status","r",stdin);
	string s; while(getline(cin, s)) if (s[2] == 'P') { cerr << s << endl; return; }
}

template<typename T> inline T read() 
{
	register T x = 0;
	register char c; register int f(1);
	while (!isdigit(c = getchar())) if (c == '-') f = -1;
	while (x = (x << 1) + (x << 3) + (c ^ 48), isdigit(c = getchar()));
	return x * f;
}

template<typename T> inline bool chkmin(T &a,T b) { return a > b ? a = b, 1 : 0; }
template<typename T> inline bool chkmax(T &a,T b) { return a < b ? a = b, 1 : 0; }

const int maxN = 5000 + 2;
const int mod = 1e9 + 7;

int n;
int ver[maxN << 1], nxt[maxN << 1], head[maxN + 2];
int dp[maxN + 2][maxN + 2], tmp[maxN + 2], size[maxN + 2], g[maxN + 2];

inline void Inc(int &x) { x < 0 ? x += mod : 0; }
inline void Dec(int &x) { x >= mod ? x -= mod : 0; }

void link(int u, int v)
{
    static int ecnt = 0;
    ver[++ecnt] = v, nxt[ecnt] = head[u], head[u] = ecnt;
}

void dfs(int u, int fa)
{
    dp[u][1] = 1;
    size[u] = 1;
    for (int i = head[u]; i; i = nxt[i])
    {
        int v = ver[i];
        if (v == fa) continue;
        dfs(v, u);
        for (int i = 0; i <= size[u]; ++i)
            for (int j = 0; j <= size[v]; ++j)
                Dec(tmp[i + j] += 1ll * dp[u][i] * dp[v][j] % mod);
        size[u] += size[v];
        for (int i = 0; i <= size[u]; ++i) dp[u][i] = tmp[i], tmp[i] = 0;
    }
    for (int i = 1; i <= size[u]; ++i) Inc(dp[u][0] -= 1ll * dp[u][i] * g[i] % mod);
}

int main() 
{
#ifndef ONLINE_JUDGE
	freopen("xhc.in", "r", stdin);
	freopen("xhc.out", "w", stdout);
#endif
    n = read<int>();
    for (int i = 1; i < n; ++i) 
    {
        int u = read<int>(), v = read<int>();
        link(u, v), link(v, u);
    }
    g[0] = 1;
    for (int i = 2; i <= n; ++i) g[i] = 1ll * g[i - 2] * (i - 1) % mod;
    dfs(1, 0);
    printf("%d\n", (mod - dp[1][0]) % mod);
	return 0;
}
posted @ 2019-11-04 20:15  茶Tea  阅读(355)  评论(0编辑  收藏  举报