poj 1192-最优连通子集解题报告

简单的树形dp，dp[i][0]表示不选i节点以i为根的子树获得的最大值。既然根节点不选，那么子节点只能选一个，否则无法保证连通。根节点若选的话则把所有状态值为正的节点都选了既为最优值

#include<stdio.h>
#include<string.h>
#define N 1005
int max(int a,int b)
{
    return a>b?a:b;
}
int dir[4][2]={0,1,0,-1,1,0,-1,0};
int head[N],t;
struct edge
{
    int v,next;
};
edge e[2*N];
int val[N];
struct point
{
    int x,y;
};
point p[N];
void add(int u,int v)
{
    e[t].v=v;
    e[t].next=head[u];
    head[u]=t++;
}
int dp[N][2];
bool used[N];
void init()
{
    memset(used,0,sizeof(used));
    memset(head,-1,sizeof(head));
    t=0;
}
void dfs(int u)
{
    used[u]=true;
    int i,v;
    dp[u][0]=0;
    dp[u][1]=val[u];
    for(i=head[u];i>=0;i=e[i].next)
    {
        v=e[i].v;
        if(!used[v])
        {
            dfs(v);
            dp[u][0]=max(dp[u][0],max(dp[v][0],dp[v][1]));
            if(dp[v][1]>0)
            dp[u][1]+=dp[v][1];
        }
    }
}
int main()
{
    int n;
    int i,j,k;
    scanf("%d",&n);
    init();
    for(i=1;i<=n;i++)
    {
        scanf("%d%d%d",&p[i].x,&p[i].y,&val[i]);
        for(j=1;j<i;j++)
        {
            for(k=0;k<4;k++)
            {
                if(p[j].x==p[i].x+dir[k][0]&&p[j].y==p[i].y+dir[k][1])
                {
                    add(i,j);
                    add(j,i);
                    break;
                }
            }
        }
    }
    dfs(1);
    printf("%d\n",max(dp[1][0],dp[1][1]));
    return 0;
}