Codevs 1378 选课

题目：

http://codevs.cn/problem/1378/

题解：
树形DP
多叉树转二叉树：

lft rit是左儿子和右儿子
 int x=lft[fa[i]];
 lft[fa[i]]=i;
 rit[i]=x;
 fa[x]=i;

DP更新
f[i][j]表示分配给以i号节点为根的子树j的得分。
i号节点与其右儿子是兄弟，i号选否与右儿子无关
i号节点与其左儿子是父子，i号选了，左儿子才能选、
因为选择了无用的根节点n+1，所以m要+1,；

for(int i=1;i<=m+1;i++){
        for(int j=0;j<i;j++){
            f[pos][i]=max(f[pos][i],f[lft[pos]][j]+w[pos]+f[rit[pos]][i-j-1]);
            //强制选择pos号，枚举分给左儿子的数量，，从0开始，因为可以只选择pos，到i-1结束
        }
        f[pos][i]=max(f[pos][i],f[rit[pos]][i]);//不选择pos，全部分给兄弟
    }

注意：
当用0号做根节点时，

f[pos][i]=max(f[pos][i],f[lft[pos]][j]+w[pos]+f[rit[pos]][i-j-1]);

pos=0时，rit[pos]=0，更新乱了，，所以会wa
感谢外校的大佬帮我指正错误

所以乖巧的用n+1做根节点好了。。。

愚蠢的矿工是两倍经验哦~~~，说好的可以挖呢？？明明是必须挖
代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=500,M=300+200;
int n,m;
int lft[N],rit[N],f[N][M],w[N],fa[N];
void dp(int pos){

    if(lft[pos]) dp(lft[pos]);//递归DP
    if(rit[pos]) dp(rit[pos]);
    for(int i=1;i<=m+1;i++){
        for(int j=0;j<i;j++){
            f[pos][i]=max(f[pos][i],f[lft[pos]][j]+w[pos]+f[rit[pos]][i-j-1]);
        }
        f[pos][i]=max(f[pos][i],f[rit[pos]][i]);
    }


}
int main(){

    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d%d",&fa[i],&w[i]);
        if(fa[i]==0) fa[i]=n+1;
        int x=lft[fa[i]];
        lft[fa[i]]=i;
        rit[i]=x;
        fa[x]=i;
    }
    dp(n+1);
    printf("%d\n",f[n+1][m+1]);
    return 0;
}