程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> HDU 1520 Anniversary party (樹形DP)

HDU 1520 Anniversary party (樹形DP)

編輯:C++入門知識

HDU 1520 Anniversary party (樹形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 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
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;
}


  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved