小红树上染色

小红树上染色_第1张图片

 记忆化深搜

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define ll long long
#define PII pair
#define TUP tuple
using namespace std;

const int N = 1e5 + 10;
const int mod = 1e9 + 7;
vector 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;
}

你可能感兴趣的:(算法题,深度优先,算法)