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 6005int 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;}