树状dp入门

Description

实验室被zser大魔王统治了，这天zser想选出一部分人看看他们对zser的恐惧值，不过实验室中除了某云飞之外每个人都有他最讨厌的人，就像树形结构一样。zser选出的人中不能存在一个人讨厌一个人。问zser选出的人可以带来的最大恐惧值为多少。

Input

第1行1个正整数 n，表示n个人(n<=6000)。
第i+1行为第i个人对zser的恐惧值ti(-1000<=ti<=1000)。
之后的n-1行每行有两个正整数x,y表示x最讨厌的人为y(x,y<=n)。
某云飞为1号。

Output

输出一行为可以zser可以得到的最大恐惧值。

Sample Input Copy

5
1 
2
3
4
5
2 1
3 1
4 2
5 2

Sample Output Copy

12

读完题也不知道要咋写，学习了树状数组才有了一点点思路；

题大概意思就是，有n个人，每个人对应有不同的价值，但有n-1种关系，并且他们不能同时存在，求输出的最大价值总和；

思路：我们先定义一个二维dp[i][0/1]表示先不选择第i个人，然后刚开始输入时都默认选择，然后是dfs建树，我们从vector第一行开始，每次搜索下面的子树，还要考虑到选不选当前树，如果不选就要看子树选不选，如果选就要+子树不选的价值，就这个亚子一直递归下去，这样就可以得到根节点的价值最大为多少；

#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<string.h>
#include<queue>
#include<vector>
using namespace std;
int dp[6010][3];
vector<vector<int> >f(6010);
void dfs(int x)
{
    int len=f[x].size();
    for(int i=0; i<len; i++)
    {
        if(len)
            dfs(f[x][i]);
        dp[x][0]=dp[x][0]+max(dp[f[x][i]][0],dp[f[x][i]][1]);
        dp[x][1]=dp[x][1]+dp[f[x][i]][0];
    }
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        for (int i = 0; i < 6010; i++)
            f[i].resize(0);
        int x,y;
        memset(dp,0,sizeof(dp));
        for(int i=1; i<=n; i++)
            scanf("%d",&dp[i][1]);
        for(int i=1; i<n; i++)
        {
            scanf("%d%d",&x,&y);
            f[y].push_back(x);
        }
        dfs(1);
        int ans=0;
        ans=max(dp[1][0],dp[1][1]);
        printf("%d\n",ans);
    }
    return 0;
}