hdu 1520 Anniversary party(基本的树形DP)

 

1、http://acm.hdu.edu.cn/showproblem.php?pid=1520

2、题目大意：

有N个人，给出每个人的权值，然后给出他们之间的关系，l k,代表l是员工，k是l的直接上司，现在要求是不让有直接关系的人一块出现在晚会上，求出满足这个要求的人的最大权值是多少。

定义dp[i][0]表示i不选，dp[i][1]表示i选

dp[i][]表示以i为根节点的子树的最大权值和

dp[i][0]=max(dp[j][0],dp[j][1]);//i选时，她的子节点可选可不选，dp[i][0]+=max(dp[j][0],dp[j][1])

dp[i][1]+=dp[j][0];,i结点选，那么他的子节点都不能选

对于这道题来说，首先得建一棵树，可以将给定的关系，存到邻接链表中，使用vector,

#include<vector>

vector<int> adj[N];

如要将v放到以u开头的链中，adj[u].push_back(v);

注意每次使用后，都清空，

for(int i=1;i<=n;i++)

adj[i].clear();

建好树之后，用dfs(),将子节点的值更新到父节点中，最终根节点中的值就是所求，max(dp[u][0],dp[u][1]);

3、AC代码：

#include<stdio.h>
#include<algorithm>
using namespace std;
#include<vector>
#include<string.h>
#define N 6005
int num[N];
vector<int> adj[N];
int indg[N];
int dp[N][2];
int visit[N];
void dfs(int v)
{
    dp[v][0]=0;
    dp[v][1]=num[v];
    visit[v]=1;
    for(int i=0;i<adj[v].size();i++)
    {
        int u=adj[v][i];
        if(visit[u])
        continue;
        dfs(u);
        dp[v][0]+=max(dp[u][1],dp[u][0]);
        dp[v][1]+=dp[u][0];
    }
}
int main()
{
    int n,a,b;
    while(scanf("%d",&n)!=EOF)
    {
        memset(indg,0,sizeof(indg));
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++)//注意每次使用清除之前的记录
        adj[i].clear();
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&num[i]);
        }
        while(scanf("%d%d",&a,&b))
        {
            if(a==0 && b==0)
            break;
            adj[b].push_back(a);
            indg[a]++;
        }
       memset(visit,0,sizeof(visit));
        for(int i=1;i<=n;i++)
        {
            if(indg[i]==0)
            {
                dfs(i);
                printf("%d\n",max(dp[i][0],dp[i][1]));
                break;
            }
        }
    }
    return 0;
}