Codevs 1378 选课

题意：
（中文）
思路：
由于父子关系特别清晰，由于每个点都只有一个前驱，并且只有n个关系，那么可以判断他们的关系是一个森林。对于每个根，如果不选择根点，此根以后的所有的点都不选择；如果选择根点，那么剩下的m - 1 - （此根代表的树选择的课程数）就分给剩下的森林来选择。
那么我们把森林转换成二叉树会更方便编程，所以我们在存储的时候就把关系变成一个二叉树来表示，左儿子表示根节点的孩子，右儿子表示根节点的兄弟。
Code：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cctype>
#include<cmath>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<bitset>
#include<queue>
#include<stack>
#include<list>
#include<map>
#include<set>

#define TEST

#define LL long long
#define Mt(f, x) memset(f, x, sizeof(f));
#define rep(i, s, e) for(int i = (s); i <= (e); ++i)
#ifdef TEST
    #define See(a) cout << #a << " = " << a << endl;
    #define See2(a, b) cout << #a << " = " << a << ' ' << #b << " = " << b << endl;
    #define debug(a, s, e) rep(_i, s, e) {cout << a[_i] << ' ';} cout << endl;
    #define debug2(a, s, e, ss, ee) rep(i_, s, e) {debug(a[i_], ss, ee)}
#else
    #define See(a)
    #define See2(a, b)
    #define debug(a, s, e)
    #define debug2(a, s, e, ss, ee)
#endif // TEST

const int MAX = 2e9;
const int MIN = -2e9;
const double eps = 1e-8;
const double PI = acos(-1.0);

using namespace std;

const int N = 305;

int s[N], tree[N][2], f[N][N];//tree[i][0]代表i节点的儿子
                              //tree[i][1]代表i节点的兄弟
//f[i][k]代表i节点选择k个课能得到的最大学分
int max3(int a, int b, int c)
{
    return max(a, max(b, c));
}

int dfs(int node, int m)
{
    if(node == 0 || m == 0) return 0;
    if(f[node][m] != -1) return f[node][m];
    for(int i = 0; i < m; ++i)
    {
        f[node][m] = max3(f[node][m],
                          dfs(tree[node][0], m - i - 1) + dfs(tree[node][1], i) + s[node],//选择根节点：此根节点的数选择m - i个课，剩下的i个分给后面的树来选择。
                          dfs(tree[node][1], m));//不选择根节点
    }
    return f[node][m];
}

int main()
{
    int n, m;
    while(~scanf("%d%d", &n, &m))
    {
        Mt(tree, 0);
        Mt(f, -1);
        for(int i = 1; i <= n; ++i)//森林转二叉树
        {
            int bef;
            scanf("%d%d", &bef, &s[i]);
            if(bef == 0) bef = n + 1;
            tree[i][1] = tree[bef][0];
            tree[bef][0] = i;
        }
        printf("%d\n", dfs(tree[n + 1][0], m));//tree[n + 1][0]存的是二叉树的根。
    }
    return 0;
}