HDU1520题解

<a target=_blank href="http://acm.hdu.edu.cn/showproblem.php?pid=1520">点击打开链接</a>

呜呜，第一道树形dp，入门水题~~~错的太傻叉了！！！→_→首先，输入要用scanf，cin超时→_→总算体会到c的强大，然后T^T要记得memset（dp）

题意很经典，n个人，每个人都有他们的权值，每个人都可以来或不来，但有一个关系，就是如果上司来了，下属就不会来→_→隐藏一个上司不来，下属随便来的条件

最终要使来的人权值之和最大

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <string>
#include <queue>
#include <stack>
#include <set>
#include <map>


using namespace std;

typedef long long ll;
typedef pair<int, int> pii;
const int N = 6005;
const int mod = 1e9 + 7;
const int inf = 0x3f3f3f3f;

int v[N];
int dp[N][2];
set<int>q[N];

void tree(int x)
{
    set<int>::iterator i;
    for(i = q[x].begin(); i != q[x].end(); i++)
    {
        int y = *i;
        tree(y);
        dp[x][1] += dp[y][0];//x来，等于所有子节点y不来
        dp[x][0] += max(dp[y][0], dp[y][1]);//x不来，子节点可来可不来
    }
    dp[x][1] += v[x];//计算完加上权值
}
int main()
{
    int n;
    while(cin >> n)
    {
        for(int i = 0; i <= n; i++)q[i].clear();//这里倒是记得初始化了
        for(int i = 1; i <= n; i++)q[0].insert(i);//意思是把所有节点都当做0的儿子，这样答案很明显就是dp[0][0]，因为0根本不存在，怎么来？所以变相就是答案
        for(int i = 1; i <= n; i++)scanf("%d", &v[i]);//输入权值
        int a, b;
        while(~scanf("%d%d", &a, &b) && (a + b))//这里由于输入个数不明确，按最坏情况应该是n*(n-1)/2，还蛮大的，反正我一开始用cin就超时了，果断scanf
        {
            if(q[0].find(a) != q[0].end())//把父亲不是0的从0那里移走
                q[0].erase(a);
            q[b].insert(a);//把a这个儿子放进父亲b里
        }
        memset(dp,0,sizeof(dp));//就是这步忘记了，→_→不然早A了
        tree(0);
        cout << dp[0][0] << endl;
    }
    return 0;
}