选课

题目描述

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

输入输出格式

输入格式：
第一行有两个整数N,M用空格隔开。(1<=N<=300,1<=M<=300)

接下来的N行,第I+1行包含两个整数ki和si, ki表示第I门课的直接先修课，si表示第I门课的学分。若ki=0表示没有直接先修课（1<=ki<=N, 1<=si<=20）。

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

输入输出样例

输入样例#1： 复制
7 4
2 2
0 1
0 4
2 1
7 1
7 6
2 2
输出样例#1： 复制
13
普通树转二叉树：把右兄弟当成右儿子
树形dp
f[i][j]:以i为根的树选了j个节点时获得的最大价值
f[i][j]=max f[son1][k1]+f[son2][k2]+…+f[son(n)][j-k1-k2-…-k(n-1)] (0=< ki<=node(i)&&Sum(ki)<=j)
最优子结构：i的所有孩子都必须在i之前求出最优解
如何保存树的信息？
孩子兄弟表示法
后来发现是实际上每个点只保存了两个值，孩子兄弟就是转成了二叉树
于是搜索就好写了
对于遍历到的每一个点，有三种决策
1、选左孩子k个
2、选右兄弟j-k-1个
3.跳过右兄弟，选下一个右兄弟

#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,num_leaf,f[310][310],s[310];
struct Tree{
    int left_child,right_brother,father;
}node[310];

int dfs(int i,int j)//i为根的树选j个节点，记忆化搜索一般和动规f数组下标的含义相同？
{
    if (j>m) return 0;
    if (f[i][j]!=-1) return f[i][j];
    for (int k=0; k<j; k++)
        f[i][j]=max(f[i][j],max(dfs(node[i].left_child,k)+dfs(node[i].right_brother,j-k-1)+s[i],dfs(node[i].right_brother,j)));
    return f[i][j];
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1; i<=n; i++) for (int j=1; j<=m; j++) f[i][j]=-1;//初始化，记忆化搜索之用 
    for (int i=1; i<=n; i++)
    {
        int fa;
        scanf("%d%d",&fa,&s[i]);
        node[i].father=fa; //f[i][1]=s[i];
    }
    for (int i=1; i<=n; i++)
    {
        int fa=node[i].father;
        if (node[fa].left_child==0)
        {
            node[fa].left_child=i;
        }
        else{
            int temp=node[fa].left_child;
            while (node[temp].right_brother) temp=node[temp].right_brother;
            node[temp].right_brother=i;
        }
    }
/*  for (int i=1; i<=n; i++)
        printf("fa:%d left:%d right:%d\n",node[i].father,node[i].left_child,node[i].right_brother);*/
    printf("%d",dfs(node[0].left_child,m));
    return 0;
}