hdu 1520(树形dp)

最新推荐文章于 2019-07-02 17:05:01 发布

weixin_30412577

最新推荐文章于 2019-07-02 17:05:01 发布

阅读量60

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Wangwanxiang/p/7380275.html

本文介绍了一种使用动态规划解决特定员工选择问题的方法。通过递归地定义子问题并结合最优子结构特性，该算法有效地计算了最大价值。文章包含完整的C++代码实现，并详细解释了每个步骤。

刚开始学，自己不服气看了看书就写，写了1个小时没解决，还是看了题解......dp[v][0]表示这个人没来，dp[v][1]表示这个人来了

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;
int n;
const int maxn=6000+10;
int val[maxn];
vector<int> g[maxn];
int dp[maxn][5];
int md[maxn];
void dfs(int x)
{
      dp[x][0]=0;
      dp[x][1]=val[x];
      for(int i=0;i<g[x].size();i++)
          {    int u=g[x][i];
               dfs(u);
              dp[x][0]+=max(dp[u][0],dp[u][1]);//老板不来，员工可以来，也可以不来
              dp[x][1]+=dp[u][0];//老板来了，员工不能来
          }
}
int main()
{
   while(~scanf("%d",&n))
   {
     memset(dp,0,sizeof(dp));
     memset(md,0,sizeof(md));
      for(int i=1;i<=n;i++)
           g[i].clear();
      for(int i=1;i<=n;i++)
        scanf("%d",&val[i]);
         int a,b;
        while(1)
        {
            scanf("%d%d",&a,&b);
            if(!a&&!b) break;
           g[b].push_back(a);
           md[a]++;
        }
    int mm=0;
       for(int i=1;i<=n;i++)
           if(!md[i])
             mm=i;
           dfs(mm);
         printf("%d\n",max(dp[mm][0],dp[mm][1]));
   }
    return 0;
}