POJ 2342 Anniversary party

2016暑期集训3-F

POJ 2342 Anniversary party

树形DP入门

传送门：HustOJ

传送门：POJ

---

题意

公司的舞会中，每个人如果看到了自己的直接上司，就会不高兴。现在给出每个人高兴时候的快乐值，和所有直接上司的从属关系。求舞会中所有人的快乐值之和的最大值。保证只有一个人没有上司!

---

思路

1. 状态
   f[i][j]表示，以i为根的子树中，i来或不来能得到的最大和。(j=0,1)

2. 转移
   f[i][0]=max{f[j][0/1]}+max{f[k][0/1]}+…(所有儿子的和)
   f[i][1]=f[j][0]+f[k][0]+…
   f[i][1]还要加上自己的快乐值。

3. 递归调用

4. 存图方法：邻接表！

5. 调用入口：用boss[i]表示i的直接上司，boss[i]=0的就是最大boss。

---

代码

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<iomanip>
#include<string>

using namespace std;
const int MAXN = 6007;
const int oo = 2000000007;
const long long int loo = 2000000000000000007ll;
typedef long double ld;
typedef long long ll;
using namespace std;
int boss[MAXN];
int happy[MAXN];
int n;
int dp[MAXN][2];
vector<int > employee[MAXN];
void dp_ontree(int n)
{
    for(int i=0;i<employee[n].size();i++)
    {
        int kk=employee[n][i];
        dp_ontree(kk);
        dp[n][0]+=max(dp[kk][0],dp[kk][1]);
        dp[n][1]+=dp[kk][0];
    }
    dp[n][1]+=happy[n];
}

int main()
{

    scanf("%d",&n);
    memset(happy,0,sizeof(happy));
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&happy[i]);
    }
    memset(boss,0,sizeof(boss));
    memset(dp,0,sizeof(dp));
    int ta,tb;
    scanf("%d%d",&ta,&tb);
    while(ta!=0&&tb!=0)
    {
        employee[tb].push_back(ta);
        boss[ta]=tb;
        scanf("%d%d",&ta,&tb);
    }
    for(int i=1;i<=n;i++)
    {
        if(boss[i]==0)
        {
            dp_ontree(i);
            int res=max(dp[i][0],dp[i][1]);
            printf("%d\n",res);
            break;
        }
    }
    return 0;
}