pku1192 最优连通子集

http://poj.org/problem?id=1192

题目大意：有一棵树，节点有整型的权值，

如果定义，树中所有节点的权值和为树的权值，题目是求权值最大的子树

#include <stdio.h>
#include <string.h>
#include <vector>

#define N 1010

using namespace std;

int n, x[N], y[N], c[N], mark[N] = {0};
vector<int> a[N];

int abs(int x)
{
    return x<0? -x: x;
}

int dfs(int x)
{
    int i, j, temp, sum = c[x];
    for(i=0; i<a[x].size(); i++)
    {
        j = a[x][i];
        if(mark[j] == 0)
        {
            mark[j] = 1;
            temp = dfs(j);
            if(temp > 0)
            {
                sum += temp;
            }
        }
    }
    return sum;
}

int main()
{
    int i, j;
    scanf("%d", &n);
    for(i=1; i<=n; i++)
    {
        scanf("%d%d%d", x+i, y+i, c+i);
    }
    for(i=1; i<n; i++)
    {
        for(j=i+1; j<=n; j++)
        {
            if(abs(x[i]-x[j]) + abs(y[i]-y[j]) == 1)
            {
                a[i].push_back(j);
                a[j].push_back(i);
            }
        }
    }
    mark[1] = 1;
    printf("%d\n", dfs(1));
    return 0;
}