学校实行学分制。每门的必修课都有固定的学分,同时还必须获得相应的选修课程学分。学校开设了N(N <
300)门的选修课程,每个学生可选课程的数量M是给定的。学生选修了这M门课并考核通过就能获得相应的学分。
在选修课程中,有些课程可以直接选修,有些课程需要一定的基础知识,必须在选了其它的一些课程的基础上才能选修。例如《Frontpage》必须在选修了《Windows操作基础》之后才能选修。我们称《Windows操作基础》是《Frontpage》的先修课。每门课的直接先修课最多只有一门。两门课也可能存在相同的先修课。每门课都有一个课号,依次为1,2,3,…。你的任务是为自己确定一个选课方案,使得你能得到的学分最多,并且必须满足先修课优先的原则。假定课程之间不存在时间上的冲突
输入 第一行有两个整数N,M用空格隔开。(1<=N<=300,1<=M<=200)。
以下N行每行代表一门课。课号依次为1,2,…,N。每行有两个数(用一个空格隔开),第一个数为这门课先修课的课号(若不存在先修课则该项为0),第二个数为这门课的学分。学分是不超过10的正整数。
输出 只有一行,选M门课程的最大得分。
样例输入
7 4
2 2
0 1
0 4
2 1
7 1
7 6
2 2
样例输出
13
分析:
背包类树形DP,这类DP的状态设计大多是第一维为节点,第二维为背包容量,结合了两类DP的状态设计方式。那么我们设
f
[
i
]
[
j
]
f[i][j]
f[i][j]为在i的子树中选j门课程的最高学分
初始化:
f
[
x
]
[
0
]
=
0
f[x][0]=0
f[x][0]=0
状态转移:
f
[
x
]
[
t
]
=
m
a
x
(
f
[
y
]
[
c
]
)
+
s
c
o
r
e
[
x
]
(
y
∈
s
o
n
[
x
]
a
n
d
Σ
c
=
t
−
1
)
f[x][t]=max(f[y][c])+score[x] (y∈son[x] and Σc=t-1)
f[x][t]=max(f[y][c])+score[x](y∈son[x]andΣc=t−1)
上面的方程还不涉及背包的内容,只是简单描述了树形结构上的转移,背包的转移和一般的背包类似,将一个子节点的f值视为价值,枚举容量,背包容量为t-1,就可以
O
(
n
2
)
O(n^2)
O(n2)做背包,然后树上的转移是
O
(
n
)
O(n)
O(n)的,所以整体复杂度为
O
(
n
3
)
O(n^3)
O(n3)
注意节点0的容量是t
代码:
#include<bits/stdc++.h>
using namespace std;
const int N=305,M=700;
int pt[N],vis[M],tot=0,head[M],nxt[M];
int f[N][N],score[N];
vector<int>son[N];
int n,m;
inline void add(int x,int y){vis[++tot]=y,nxt[tot]=head[x],head[x]=tot;}
void inson(int v){
pt[v]=1;
for(int i=head[v];i;i=nxt[i]){
if(pt[vis[i]]) continue;
son[v].push_back(vis[i]);
inson(vis[i]);
}
}
void dfs(int v){
f[v][0]=0;
for(int i=0;i<son[v].size();i++){
int y=son[v][i];
dfs(y);
for(int t=m;t>=0;t--){
for(int j=t;j>=0;j--){
if(t-j>=0)
f[v][t]=max(f[v][t],f[v][t-j]+f[y][j]);
}
}
}
if(v!=0)
for(int t=m;t;t--)
f[v][t]=f[v][t-1]+score[v];
}
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++){
int x,y;
scanf("%d%d",&x,&score[i]);
add(i,x);
add(x,i);
}
inson(0);
dfs(0);
int ans=0;
for(int i=m;i>=0;i--) ans=max(ans,f[0][i]);
cout<<ans<<endl;
return 0;
}