题目描述
在大学里每个学生,为了达到一定的学分,必须从很多课程里选择一些课程来学习,在课程里有些课程必须在某些课程之前学习,如高等数学总是在其它课程之前学习。现在有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
f[i][j] 第i个节点选j个点的最大收益
dfs的同时dp,保证顺序
#include<iostream>
#include<cstdio>
using namespace std;
const int MAXN=1005;
inline int read_d()
{
int s=0;
char c;
while(c=getchar(),c<'0'||c>'9');
while(c<='9'&&c>='0') {
s=s*10+c-'0';
c=getchar();
}
return s;
}
int n,m;
int V[MAXN];
struct Edge {
int next,to,w;
} e[MAXN];
int ecnt=1,head[MAXN];
inline void add(int x,int y,int w)
{
e[++ecnt].to = y;
e[ecnt].next = head[x];
e[ecnt].w =w;
head[x]=ecnt;
}
int f[MAXN][MAXN];
int dfs(int now)
{
int scnt=0;//
for(int i=head[now]; i; i=e[i].next) {
int v=e[i].to ;
int tmp=dfs(v);
scnt+=tmp+1;
for(int j=scnt; j>=0; j--) {
for(int k=0; k<=tmp; k++) {
if(j-k-1<0) continue;
f[now][j]=max(f[now][j],f[now][j-k-1]+f[v][k]+e[i].w);
}
}
}
return scnt;
}
int main()
{
// freopen("work.in","r",stdin);
scanf("%d%d",&n,&m);
for(int i=1; i<=n; i++) {
int x;
scanf("%d%d",&x,&V[i]);
add(x,i,V[i]);
}
dfs(0);
printf("%d
",f[0][m]);
return 0;
}