码迷,mamicode.com
首页 > 其他好文 > 详细

HDU 1520 Anniversary party (树形DP)

时间:2015-01-16 22:28:50      阅读:172      评论:0      收藏:0      [点我收藏+]

标签:acm   c语言   算法   编程   树形dp   

树形DP的关键在于如何处理递归返回的信息。这题dp[i][0]表示不选i点时当前最高权值。dp[i][1]表示选i点时当前最高权值。状态转移方程:dp[u][[0]+=max(dp[v][0],dp[v][1]),dp[u][1]+=dp[v][0];

代码如下:

#include <iostream>
#include <string.h>
#include <math.h>
#include <queue>
#include <algorithm>
#include <stdlib.h>
#include <map>
#include <set>
#include <stdio.h>
using namespace std;
#define LL __int64
#define pi acos(-1.0)
const int mod=100000000;
const int INF=0x3f3f3f3f;
const double eqs=1e-8;
int dp[7000][2], c[7000], vis[7000];
int head[7000], cnt, n;
struct node
{
    int u, v, next;
}edge[100000];
void add(int u, int v)
{
    edge[cnt].v=v;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}
void dfs(int u)
{
    vis[u]=1;
    dp[u][1]=c[u];
    for(int i=head[u];i+1;i=edge[i].next){
        int v=edge[i].v;
        if(!vis[v]){
            dfs(v);
            dp[u][0]+=max(dp[v][0],dp[v][1]);
            dp[u][1]+=dp[v][0];
        }
    }
}
void init()
{
    memset(vis,0,sizeof(vis));
    memset(head,-1,sizeof(head));
    cnt=0;
    memset(dp,0,sizeof(dp));
}
int main()
{
    int i, j, u, v;
    while(scanf("%d",&n)!=EOF){
        for(i=1;i<=n;i++){
            scanf("%d",&c[i]);
        }
        init();
        while(scanf("%d%d",&u,&v)!=EOF&&u&&v){
            add(u,v);
            add(v,u);
        }
        dfs(1);
        printf("%d\n",max(dp[1][0],dp[1][1]));
    }
    return 0;
}


HDU 1520 Anniversary party (树形DP)

标签:acm   c语言   算法   编程   树形dp   

原文地址:http://blog.csdn.net/scf0920/article/details/42783385

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!