小红树上染色

 记忆化深搜

#include <iostream>
#include <string>
#include <stack>
#include <vector>
#include <queue>
#include <deque>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <numeric>
#define ll long long
#define PII pair<int, int>
#define TUP tuple<ll, ll, ll>
using namespace std;

const int N = 1e5 + 10;
const int mod = 1e9 + 7;
vector<int> edge[N];
int f[N][2] = { 0 };
ll dfs(int node, int c, int fa) {
    ll ret = 1;
    for (auto x : edge[node]) {
        if (x == fa) continue;
        if (c == 0) {
            if (!f[x][1]) f[x][1] = dfs(x, 1, node);
            ret *= dfs(x, 1, node);
        }
        else {
            if (!f[x][1]) f[x][1] = dfs(x, 1, node);
            if (!f[x][0]) f[x][0] = dfs(x, 0, node);
            ret *= f[x][1] + f[x][0];
        }
        ret %= mod;
    }
    return ret;
}
int main() {
    int n;
    cin >> n;
    for (int i = 1; i < n; i++) {
        int u, v;
        cin >> u >> v;
        edge[u].push_back(v);
        edge[v].push_back(u);
    }
    ll ans = dfs(1, 0, -1) + dfs(1, 1, -1);
    cout << ans % mod;
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

云儿乱飘

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值