【例题】【动规(树形)】NKOJ 1217 选课、NKOJ 2317 选课2.0

NKOJ 1217 选课
时间限制 : 10000 MS 空间限制 : 65536 KB

问题描述
在大学里每个学生，为了达到一定的学分，必须从很多课程里选择一些课程来学习，在课程里有些课程必须在某些课程之前学习，如高等数学总是在其它课程之前学习。现在有N门功课，每门课有个学分，每门课最多只有一个直接先修课，每门课可是最多两门其它课程的直接先修课（若课程a是课程b的先修课即只有学完了课程a，才能学习课程b）。一个学生要从这些课程里选择M门课程学习，问他能获得的最大学分是多少？

输入格式
第一行有两个整数N,M用空格隔开。(1<=N<=200,1<=M<=150)
接下来的N行,第I+1行包含两个整数ki和si, ki表示第i门课的直接先修课，si表示第I门课的学分。若ki=0表示没有直接先修课（1<=ki<=N, 1<=si<=20）。数据保证只有一门课程ki=0

输出格式
只有一行，选M门课程的最大得分。

样例输入
5 3
3 5
5 18
5 2
3 10
0 7

样例输出
27

思路
问题实际为：在一个N个结点的二叉树中选取M个结点，使得所选结点的权值之和最大。
(结论)树形动规套路：
1、分为左右两棵子树，左边子树取X个节点
右边子树取Y个节点使得 X+Y+1=M
2、对于划分出来的子树，可按同样的方法再递归划分为左X1,右Y1两棵子树，使得X1+Y1+1=Y
综上，步骤为：
建树….二叉
阶段：每节点所代表的子树
状态：以每棵子树取得的最优值
决策：当前节点的最优值由其子节点代表的子树的最优值选择而来
实现：递归程序实现DP（记忆化搜索）

对于该题：
阶段：依次讨论以每个节点为根的子树。
状态：f[x][y]表示以x为根的子树选出y个节点的最大值。
决策：x的左右两棵子树分别保留多少个节点。
方程：f[x][y]=max{ f[le[x]][k]+f[ri[x]][y-1-k]+v[x] }
边界：1<=x<=n ，1<=y<=m ，0<=k<=y-1

代码：

#include<cstdio>
#include<iostream>
using namespace std;

int le[203],ri[203],s[203],f[203][203];
int n,m,root;

void dp_(int x,int y)
{
    if(f[x][y]>0) return ;
    if(x!=0)
    {
        for(int k=0;k<y;k++)
        {
            dp_(le[x],k),dp_(ri[x],y-k-1);
            int now=f[le[x]][k]+f[ri[x]][y-k-1]+s[x];
            f[x][y]=max(now,f[x][y]);
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    int a,b;
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&a,&b);
        s[i]=b;
        if(a!=0) le[a]==0?le[a]=i:ri[a]=i;
        else root=i;
    }
    dp_(root,m);
    printf("%d",f[root][m]);
}

变式：若权值在边上，则仍认为权值在点上

NKOJ 2317 选课2.0
时间限制 : 10000 MS 空间限制 : 65536 KB

问题描述
学校实行学分制。每门的必修课都有固定的学分，同时还必须获得相应的选修课程学分。学校开设了N（N<300）门的选修课程，每个学生可选课程的数量M是给定的。学生选修了这M门课并考核通过就能获得相应的学分。
　　在选修课程中，有些课程可以直接选修，有些课程需要一定的基础知识，必须在选了其它的一些课程的基础上才能选修。例如《Frontpage》必须在选修了《Windows操作基础》之后才能选修。我们称《Windows操作基础》是《Frontpage》的先修课。每门课的直接先修课最多只有一门。两门课也可能存在相同的先修课。每门课都有一个课号，依次为1，2，3，…。 　　
你的任务是为自己确定一个选课方案，使得你能得到的学分最多，并且必须满足先修课优先的原则。假定课程之间不存在时间上的冲突。

输入格式
第一行包括两个整数N、M（中间用一个空格隔开）其中1≤N≤300,1≤M≤N。
以下N行每行代表一门课。课号依次为1，2，…，N。
每行有两个数（用一个空格隔开），第一个数为这门课先修课的课号（若不存在先修课则该项为0），第二个数为这门课的学分。
学分是不超过10的正整数。

输出格式
只有一个数，实际所选课程的最大学分总数。

样例输入
7 4
2 2
0 1
0 4
2 1
7 1
7 6
2 2

样例输出
13

思路：
问题为：在一个N个结点的森林中选取M个结点使得所选结点的权值之和最大，同时满足每次选取时，若选儿子结点，必选根结点的条件
方法：同上，但是要先转二叉树，并注意满足若选儿子结点，必选根结点的条件。
/* 多叉树转二叉树：http://blog.csdn.net/y__xv/article/details/51657116*/
阶段、状态、决策..
方程：f[i][j]=max{f[left[i]][k]+f[right[i]][j-k-1]+v[i]，
f[right[i]][j]}
//左孩子为根节点原孩子，右孩子为根节点原兄弟，所以左右情况不同,若左子树有被选，则根节点也必须选；若右子树被选，则根节点可选可不选。
边界：1<=i<=n , 0<=k< j<.n

代码：

**#include<cstdio>
#include<iostream>
using namespace std;

int n,m,root;
int s[303],ri[303],le[303],f[303][303];

void dp_(int x,int y)
{
    if(f[x][y]) return;
    if(x==0) return;
    int temp=0;
    for(int i=0;i<y;i++)
    {
        dp_(ri[x],i),dp_(le[x],y-i-1);
        temp=max(f[le[x]][y-i-1]+f[ri[x]][i]+s[x],temp);
    }
    dp_(ri[x],y);
    f[x][y]=max(temp,f[ri[x]][y]);
}

int main()
{
    scanf("%d%d",&n,&m);
    root=n+1;
    int a,b;
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&a,&b);
        s[i]=b;
        if(a==0) a=root;
        ri[i]=le[a];
        le[a]=i;
    }
    dp_(root,m+1);
    printf("%d",f[root][m+1]);
}**