【树形DP】wikioi 1380 没有上司的舞会

题目来源:http://wikioi.com/problem/1380/

          https://www.tyvj.cn/Problem_Show.aspx?id=1052

分析:

树状动态规划

对于每个人,都有去和不去两种可能,所以找到根节点,向下处理。

对于每个节点 

 1、如果到场,则在这个点的气氛值 =其所有下属都不到场时的气氛值 +其本身的气氛值。

 2、如果不到场,则在这个点的气氛值=其下属到场(或者不到场,二者取较大值)时的气氛值。

就这样递归求解,最后输出根节点的气氛值即可。

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
vector<int> g[6000];                       //用vector存储每个节点的下属 
int f[60000][2],a[6000],b[6000],n;
void dfs(int k)                             //递归求解 
{
     for (int i=0;i<g[k].size();i++)       //搜索每一个子节点 
     {
         int h=g[k][i];
         dfs(h);
         f[k][0]+=max(f[h][1],f[h][0]);      //取或者不取,选择较大值 
         f[k][1]+=f[h][0];         
     }
     f[k][1]+=a[k];
}

int main()
{    
    scanf("%d",&n);
    for (int i=1;i<=n;i++) 
    scanf("%d",&a[i]);
    for (int i=1;;i++) 
     {
             int x,y;
             scanf("%d%d",&x,&y);
             if (x==0&&y==0) break;
             g[y].push_back(x);            //储存y的子节点 
             b[x]=y;                        //储存x的父节点 
     }
    for (int i=1;i<=n;i++)
     {
        if (b[i]==0)                //寻找根节点 
          {
              dfs(i);
              printf("%d",max(f[i][0],f[i][1]));
              break;
          }
     }
     return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值