2021 ICPC 区域赛(上海) G.Edge Groups（树形dp）

阐上

已于 2022-02-15 13:12:31 修改

阅读量657

点赞数

分类专栏： # 树形dp 文章标签：动态规划算法树结构 c++ 组合数学

于 2022-02-14 18:22:17 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_51797626/article/details/122929837

版权

树形dp 专栏收录该内容

12 篇文章

订阅专栏

边集合

在这里插入图片描述
题目大意

给出一个点数为n的树（n为奇数），将n-1条边两两分组。每组内需要满足：有两条边，且这两条边要有一个公共点。问有多少种方案。

引用 lwz_159的题解博客

$C o d e :$

#include<bits/stdc++.h>
#include<unordered_map>
#define mem(a,b) memset(a,b,sizeof a)
#define cinios (ios::sync_with_stdio(false),cin.tie(0),cout.tie(0))
#define sca scanf
#define pri printf
#define forr(a,b,c) for(int a=b;a<=c;a++)
#define rfor(a,b,c) for(int a=b;a>=c;a--)
#define endl "\n"
//[博客地址]：https://blog.youkuaiyun.com/weixin_51797626?t=1
using namespace std;
inline void read(int& x) { x = 0; int f = 1; char ch = getchar(); while (ch < '0' || ch > '9') { if (ch == '-') f = -1; ch = getchar(); } while (ch >= '0' && ch <= '9') { x = x * 10 + (ch - '0'); ch = getchar(); } x *= f; }
void write(int x) { if (x < 0) putchar('-'), x = -x; if (x >= 10) write(x / 10); putchar(x % 10 + '0'); }

typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> PII;

const int N = 100010, M = 200010, MM = N;
int INF = 0x3f3f3f3f, mod = 998244353;
ll LNF = 0x3f3f3f3f3f3f3f3f;
int n, m, k, T, S, D;
int h[N], ne[M], e[M], idx;
ll dp[N];

inline void add(int a, int b) {
    e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

//根据子节点所在子树的边数判断能贡献的边
//如果边数为偶数（自身内部两两配对），当前结点与子节点的边就贡献出来了
//否则奇数必定会占用这条边

bool dfs(int x, int fa) {
    int cnt = 0;
    dp[x] = 1;

    for (int i = h[x]; ~i; i = ne[i]) {
        int j = e[i];
        if (j == fa)continue;
        if (!dfs(j, x))cnt++;
        dp[x] = (dp[x] * dp[j]) % mod;
    }
    for (int i = 1; i <= cnt; i += 2)dp[x] = (dp[x] * i) % mod;
    return cnt & 1;
}

int main() {
    cinios;

    cin >> n;
    mem(h, -1);

    forr(i, 1, n - 1) {
        int a, b;
        cin >> a >> b;
        add(a, b), add(b, a);
    }
    dfs(1, -1);

    cout << dp[1];

    return 0;
}
/*
*/