题目描述
在大学里每个学生,为了达到一定的学分,必须从很多课程里选择一些课程来学习,在课程里有些课程必须在某些课程之前学习,如高等数学总是在其它课程之前学习。现在有 NN 门功课,每门课有个学分,每门课有一门或没有直接先修课(若课程 a 是课程 b 的先修课即只有学完了课程 a,才能学习课程 b)。一个学生要从这些课程里选择 MM 门课程学习,问他能获得的最大学分是多少?
输入格式
第一行有两个整数 NN , MM 用空格隔开。( 1 \leq N \leq 3001≤N≤300 , 1 \leq M \leq 3001≤M≤300 )
接下来的 NN 行,第 I+1I+1 行包含两个整数 k_iki和 s_isi, k_iki 表示第I门课的直接先修课,s_isi 表示第I门课的学分。若 k_i=0ki=0 表示没有直接先修课(1 \leq {k_i} \leq N1≤ki≤N , 1 \leq {s_i} \leq 201≤si≤20)。
输出格式
只有一行,选 MM 门课程的最大得分。
输入输出样例
输入 #1
7 4 2 2 0 1 0 4 2 1 7 1 7 6 2 2
输出 #1
13
解决代码:
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
const int N = 700;
int f[N][N];
int e[N],ne[N],w[N],h[N],idx;
int n,m;
void add(int a,int b)
{
e[idx] = b;
ne[idx] = h[a];
h[a]= idx++;
}
void dfs(int u) //处理以u为根的树,f[u][k]表示以u为根的树,修k门课的最大值
{
for(int i=h[u];i!=-1;i=ne[i])
{
int j = e[i];
dfs(j);
for(int k=m;k>=0;k--)
{
for(int s=1;s<=k;s++)
{ // f[u][k-s]表示j的兄弟节点修k-s门课
f[u][k] = max(f[u][k],f[u][k-s]+f[j][s]);
}
}
}
for(int i=m;i>=1;i--) f[u][i] = f[u][i-1] + w[u];
}
int main()
{
memset(h,-1,sizeof h);
cin>>n>>m;
for(int i=1;i<=n;i++)
{
int a;
cin>>a>>w[i];
add(a,i);
}
m++;
dfs(0);
cout<<f[0][m];
return 0;
}