题目
在N门选修课中选择M门选修课,选修课可能需要学习先修课,求可得到的最大学分。
分析
用树形dp,
f
[
x
]
[
v
]
f[x][v]
f[x][v]表示以x为根的子树中选修v门课的最大学分,
f
[
x
]
[
v
]
=
m
a
x
∑
i
=
1
n
u
m
c
i
=
v
−
1
(
f
[
s
o
n
]
[
c
i
]
)
+
s
c
o
r
e
[
x
]
f[x][v]=max_{\sum^{num}_{i=1} c_i=v-1}(f[son][c_i])+score[x]
f[x][v]=max∑i=1numci=v−1(f[son][ci])+score[x]
可以把0当成虚拟的选修课,从0开始树形dp找出答案。
代码
#include <cstdio>
#include <cctype>
using namespace std;
struct node{int y,next;}e[301];
int f[301][301],ls[301],n,m,score[301];
int in(){
int ans=0; char c=getchar();
while (!isdigit(c)) c=getchar();
while (isdigit(c)) ans=ans*10+c-48,c=getchar();
return ans;
}
int max(int a,int b){return (a>b)?a:b;}
void dp(int x){
for (int i=ls[x];i;i=e[i].next){
dp(e[i].y);
for (int v=m;v>=1;v--)
for (int j=v-1;j>=1;j--)
f[x][v]=max(f[x][v],f[x][v-j]+f[e[i].y][j]);
}
}
int main(){
n=in(); m=in()+1;
for (int i=1;i<=n;i++){
int x,y; x=in(); f[i][1]=in();
e[i]=(node){i,ls[x]}; ls[x]=i;
}
dp(0); return !printf("%d",f[0][m]);
}
PS:JZOJ 3418
#include <cstdio>
#include <cctype>
#include <vector>
#include <algorithm>
using namespace std;
struct node{int y,next;}e[1001]; vector<int>a;
int f[1001][1001],ls[1001],n,m,g[1001][1001][2];
int in(){
int ans=0; char c=getchar();
while (!isdigit(c)) c=getchar();
while (isdigit(c)) ans=ans*10+c-48,c=getchar();
return ans;
}
int dp(int x){
int site=0;
for (int i=ls[x];i;i=e[i].next){
site+=dp(e[i].y);
for (int v=site+1;v;v--)
for (int j=v-1;j;j--)
if (f[x][v-j]+f[e[i].y][j]>f[x][v])
f[x][v]=f[x][v-j]+f[e[i].y][j],g[x][v][0]=e[i].y,g[x][v][1]=j;
}
return site+1;
}
void getans(int x,int v){
a.push_back(x);
while (g[x][v][0]){
getans(g[x][v][0],g[x][v][1]);
v-=g[x][v][1];
}
}
int main(){
n=in(); m=in()+1;
for (int i=1;i<=n;i++){
int x,y; x=in(); f[i][1]=in();
e[i]=(node){i,ls[x]}; ls[x]=i;
}
dp(0); printf("%d\n",f[0][m]);
if (m<101){
getans(0,m);
stable_sort(a.begin(),a.end());
for (int i=1;i<a.size();i++) printf("%d\n",a[i]);
}
return 0;
}