hdoj 1520 Anniversary party(树形dp)

简介: 我们可以把一个节点当做一个人,每个节点都有一个权重。按照题目意思,如果我们取了某个节点,那么他的父节点和子节点都是不能取的。按要求选取节点,使得选取节点的权重和最大。

题目链接


  按照等级我们可以建一颗树,如图


cb91ae634490d69a0d3925f53c92fc13_1366961765_2156.PNG

     我们可以把一个节点当做一个人,每个节点都有一个权重。按照题目意思,如果我们取了某个节点,那么他的父节点和子节点都是不能取的。按要求选取节点,使得选取节点的权重和最大。


    DP,用no表示不选择i点时,i点及其子树能选出的最多人数,is表示选择i点时,i点及其子树的权值和最大。


状态转移方程:



—对于叶子节点 dp[k].no = 0, dp[k].no = a[k].v

—对于非叶子节点i,

—dp[i].no = ∑max(dp[j][0], dp[j][1]) (j是i的儿子)

—dp[i].is = v + ∑dp[j].no (j是i的儿子)

代码:

#include <stdio.h>
#include <algorithm>
#include <vector>
#include <string.h>
using namespace std;
struct node
{
    int v;
    int is, no;
    vector<int> son;
}a[6010];
bool f[6010];
void dp(int x)
{
    int t;
    for (unsigned int i = 0; i < a[x].son.size(); i++)
    {
        t = a[x].son[i];
        dp(t);
        a[x].is += a[t].no;
        a[x].no += max(a[t].no, a[t].is);
    }
}
int main()
{
    int n;
    while (scanf("%d", &n) != EOF)
    {
        for (int i = 1; i <= n; i++)
        {
            scanf("%d", &a[i].v);
            a[i].son.clear();
            a[i].is = a[i].v;
            a[i].no = 0;
        }
        int son, fa;
        memset(f, 0, sizeof(f));
        while (scanf("%d %d",&son, &fa) && son+fa)
        {
            a[fa].son.push_back(son);
            f[son] = 1;
        }
        for (int i = 1; i <= n; i++)
        {
            if (f[i] == false)
            {
                dp(i);
                printf("%d\n",max(a[i].is, a[i].no));
                break;
            }
        }
    }
    return 0;
}
目录
相关文章
|
算法
light oj 1258 - Making Huge Palindromes(KMP)
ight oj里这个题目是属于KMP分类的,但乍看好像不是kmp,因为只有一个字符串。要想的到一个回文串,把该字符串翻转接到原串后面必然是一个回文串,但并不一定是最短的。我们必须考虑怎么把两个串尽量融合在一起,这就要看翻转串的前段与原串的后段有多少是匹配的了,这里就用到了KMP算法。
36 1
hdoj 1028/poj 2704 Pascal's Travels(记忆化搜索||dp)
有个小球,只能向右边或下边滚动,而且它下一步滚动的步数是它在当前点上的数字,如果是0表示进入一个死胡同。求它从左上角到右下角到路径数目。 注意, 题目给了提示了,要用64位的整数。
37 0
|
Perl
AtCoder Beginner Contest 217 F - Make Pair (区间dp)
AtCoder Beginner Contest 217 F - Make Pair (区间dp)
122 0
|
Java
HDOJ(HDU) 2164 Rock, Paper, or Scissors?
HDOJ(HDU) 2164 Rock, Paper, or Scissors?
115 0
HDOJ(HDU) 1678 Shopaholic
HDOJ(HDU) 1678 Shopaholic
107 0
HDOJ/HDU 1241 Oil Deposits(经典DFS)
HDOJ/HDU 1241 Oil Deposits(经典DFS)
87 0
HDOJ(HDU) 2317 Nasty Hacks(比较、)
HDOJ(HDU) 2317 Nasty Hacks(比较、)
105 0
|
Web App开发 Java 数据安全/隐私保护
HDOJ(HDU) 1563 Find your present!(异或)
HDOJ(HDU) 1563 Find your present!(异或)
236 0
|
机器学习/深度学习
HDOJ 1334 Perfect Cubes(暴力)
HDOJ 1334 Perfect Cubes(暴力)
104 0
HDOJ1002题A + B Problem II,2个大数相加
HDOJ1002题A + B Problem II,2个大数相加
115 0