1. 程式人生 > >POJ-1192-最優連通子集

POJ-1192-最優連通子集

本來書上是放在最短路徑那章的,寫了半天發現很麻煩,一搜索才知道這個題應該用樹形DP做,

算是比較簡單的樹形DP題吧。

程式碼:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstdlib>
#include<queue>
using namespace std;
const int maxn=1001;
const int inf=1<<29;
int n,py[maxn],px[maxn],c[maxn],dp[maxn][2];
int e,pnt[maxn*maxn],nxt[maxn*maxn],head[maxn];
void addedge(int u,int v)
{
    pnt[e]=v;nxt[e]=head[u];head[u]=e++;
}
void DFS(int u,int f)
{
    dp[u][0]=0;
    dp[u][1]=c[u];
    for(int i=head[u];i!=-1;i=nxt[i])
    {
	int v=pnt[i];
	if(v!=f)
	{
	    DFS(v,u);
	    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()
{
    while(~scanf("%d",&n))
    {
	e=0;
	memset(head,-1,sizeof(head));
	memset(dp,0,sizeof(dp));
	for(int i=1;i<=n;i++)
	{
	    scanf("%d%d%d",&px[i],&py[i],&c[i]);
	    for(int j=1;j<i;j++)
		if(abs(px[i]-px[j])+abs(py[i]-py[j])==1)
		{
		    addedge(j,i);
		    addedge(i,j);
		}
	}
	DFS(1,-1);
	printf("%d\n",max(dp[1][0],dp[1][1]));
    }
    return 0;
}