最優連通子集 POJ - 1192(樹形dp)

無向連通樹的最大加權和
dp[x][0]=max(dp[x][0],max(dp[y][0],dp[y][1]));
dp[x][1]+=max(dp[y][1],0);

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <string>
#include <vector>
#define ll long long

using namespace std;
int n,dp[1005][2],vis[1005];
struct node
{
    int x,y,w;
}a[1005];
vector<int> v[1005];

void dfs(int x)
{
   dp[x][0]=0;
   dp[x][1]=a[x].w;
   vis[x]=1;
   for(int i=0;i<v[x].size();i++)
   {
       int y=v[x][i];
       if(vis[y])
            continue;
       dfs(y);
       dp[x][0]=max(dp[x][0],max(dp[y][0],dp[y][1]));
       dp[x][1]+=max(dp[y][1],0);
   }
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    while(cin >> n)
    {
        for(int i=1;i<=n;i++)
            v[i].clear();
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=n;i++)
            cin >> a[i].x >> a[i].y >> a[i].w;
        for(int i=1;i<n;i++)
        {
            for(int j=i+1;j<=n;j++)
            {
                if(abs(a[i].x-a[j].x)+abs(a[i].y-a[j].y)==1)
                {
                    v[i].push_back(j);//無向樹
                    v[j].push_back(i);
                }
            }
        }
        dfs(1);
        cout << max(dp[1][0],dp[1][1]) << endl;
    }
    return 0;
}