题目描述
在大学里每个学生,为了达到一定的学分,必须从很多课程里选择一些课程来学习,在课程里有些课程必须在某些课程之前学习,如高等数学总是在其它课程之前学习。现在有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;
}