LightOJ - 1201 A Perfect Murder(树形DP)

题目大意：有N只蚊子，蚊子之间有相应的关系，假设蚊子为点，关系为边，则蚊子可以组成K棵无根树，注意，不是1棵
现在你的任务是干掉蚊子，但是有一个限制，杀了该蚊子后，和该蚊子直连的蚊子都不能杀，在这种情况下，最多杀几只蚊子

解题思路：树形DP，用dp[i][0]表示不杀该蚊子，还能杀多少只蚊子
dp[i][1]表示杀掉该蚊子后，还能杀多少只蚊子
则dp[i][0] += max(dp[j][0], dp[j][1])，不杀该蚊子，则和其相连的蚊子就可杀，也可不杀了
dp[i][1] += dp[i][0]，杀了该蚊子，和其相连的就都不能杀了

#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
const int N = 1010;
vector<int> Tree[N];
int n, m, cas = 1;
int dp[N][2], f[N];
bool vis[N];

int find(int x) {
    return x == f[x] ? x : f[x] = find(f[x]);
}

void init() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++) {
        Tree[i].clear();
        f[i] = i;
    }

    int u, v;
    for (int i = 1; i <= m; i++) {
        scanf("%d%d", &u, &v);
        Tree[u].push_back(v);
        Tree[v].push_back(u);
        f[find(u)] = find(v);
    }
}

void dfs(int u, int fa) {
    //不杀
    dp[u][0] = 0;
    //杀
    dp[u][1] = 1;

    for (int i = 0; i < Tree[u].size(); i++) {
        int v = Tree[u][i];
        if (v == fa) continue;
        dfs(v, u);
        //不杀
        dp[u][0] += max(dp[v][0], dp[v][1]);
        //杀
        dp[u][1] += dp[v][0];
    } 
}

void solve() {
    memset(vis, 0, sizeof(vis));
    int ans = 0;
    for (int i = 1; i <= n; i++) {
        if (!vis[find(i)]) {
            dfs(i, -1);
            ans += max(dp[i][0], dp[i][1]);
            vis[find(i)] = true;
        }
    }
    printf("Case %d: %d\n", cas++, ans);
}

int main() {
    int test;
    scanf("%d", &test);
    while (test--) {
        init();
        solve();
    }
    return 0;
}