POJ 1192 最优连通子集 （树形dp）

题解 ： 对于这种 在无向图 上求最大值的题目，很多情况下都是利用树形dp 去做这种题，这个也不例外还有一个关键的地方就是这个题目保证了这一定可以看作是一棵树，并且树的形态（谁是根 ， 谁是叶子）不影响答案，只是树的边之间的关系 （哪些之间可以连边，哪些不可以）影响答案的话树形dp 值得考虑 或者一些树上的其他算法都可以。

dp (i,j) (j 只取 0、1) 表示第 i 个节点选择还是没选择的最大权值
dp (i,1) = c[i] + dp (son,1); (因为要求连通所以 不能选择 dp (son,0));
dp (i,0) = max (dp(son,1),dp(son,0)) 就是从儿子中找一个最大的就可以。
注意这个dp 是从叶子到根的树形dp 所以我们要先dfs 下去再进行dp;

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

using namespace std;
const int maxn = 1005;

struct node {
    int x,y,c;
}a[maxn];

int dp[maxn][2] = {0};
int vis[maxn] = {0};
vector <int> G[maxn];
bool check (node a,node b) {
    if (abs (a.x - b.x) + abs (a.y - b.y) <= 1) return true;
    return false;
}
void dfs (int root) {
    vis[root] = 1;
    int cnt = G[root].size();
    if (cnt == 0) {
        dp[root][1] = a[root].c;
        dp[root][0] = 0;
        return ;
    }
    dp[root][1] = a[root].c;
    dp[root][0] = 0;
    for (int i = 0;i < cnt ; ++ i) {
        int son = G[root][i];
        if (vis[son]) continue;
        dfs (son);
        if (dp[son][1] >= 0) {
            dp[root][1] += dp[son][1];
        }
        dp[root][0] = max (dp[root][0],dp[son][1]);
        dp[root][0] = max (dp[root][0],dp[son][0]);
    }
}

int main () {
    ios_base :: sync_with_stdio(false);
    int n;
    while (cin >> n) {
        memset (vis,0,sizeof (vis));
        for (int i = 1;i <= n; ++ i) {
            cin >> a[i].x >> a[i].y >> a[i].c;
        }
        for (int i = 1;i <= n; ++ i) {
            for (int j = i + 1;j <= n; ++ j) {
                if (check(a[i],a[j])) {
                    G[i].push_back (j);
                    G[j].push_back (i);
                }
            }
        }
        dfs (1);
        int ans = max (dp[1][1],dp[1][0]);
        cout << ans << endl;
    }
    return 0;
}