原题:
题目描述
在大学里每个学生,为了达到一定的学分,必须从很多课程里选择一些课程来学习,在课程里有些课程必须在某些课程之前学习,如高等数学总是在其它课程之前学习。现在有 N 门功课,每门课有个学分,每门课有一门或没有直接先修课(若课程 a 是课程 b 的先修课即只有学完了课程 a,才能学习课程 b)。一个学生要从这些课程里选择 M 门课程学习,问他能获得的最大学分是多少?
输入格式
第一行有两个整数 N , M 用空格隔开。( 1≤N≤300 , 1≤M≤300 )
接下来的 N 行,第 I+1 行包含两个整数 ki和 si, ki 表示第I门课的直接先修课,si 表示第I门课的学分。若 ki=0k_i=0ki=0 表示没有直接先修课(1≤ki≤N 1≤si≤20)。
输出格式
只有一行,选 MMM 门课程的最大得分。
输入输出样例
输入 #1
7 4
2 2
0 1
0 4
2 1
7 1
7 6
2 2
输出 #1
13
代码:
#include <bits/stdc++.h>
using namespace std;
const int maxn = 301;
int dp[maxn][maxn];
int s[maxn];
vector<int> G[maxn];
int n,m;
int dfs(int cur)
{
int node_cnt = 1;
dp[cur][1] = s[cur];
if (G[cur].empty())
{
return 1;
}
for (int i = 0; i < G[cur].size(); i++)
{
int cnt = dfs(G[cur][i]);
for (int j = min(m + 1, node_cnt); j>0; j--)
{
for (int k = 1; k <= cnt; k++)
{
if (k + j > m + 1)
{
break;
}
dp[cur][k + j] = max(dp[cur][k + j], dp[G[cur][i]][k] + dp[cur][j]);
}
}
node_cnt += cnt;
}
return node_cnt;
}
int main()
{
ios::sync_with_stdio(false);
while(cin>>n>>m)
{
int k;
for (int i=0;i<=n;i++)
{
G[i].clear();
}
memset(s, 0 ,sizeof(s));
for (int i=1;i<=n;i++)
{
cin >> k >> s[i];
G[k].push_back(i);
}
dfs(0);
cout << dp[0][m+1]<<endl;
}
return 0;
}
思路:
树形背包的裸题,也是树形dp类型的入门题目。
树形dp,肯定是以一个子树作为一个子状态去考虑,这个问题也显然是这样的。
从背包模型上来看,物品的价值是课程的分数,物品的体积是1,代表一门课程。
从树形结构来考虑,有个天然的子结构,那就是一颗子树。
与正常的背包模型不同的是,如果考虑以u为根节点的子树,如果要求考虑其子节点v,如果要将v的状态合并到u中,那么也要将v的子孙节点一并合入到u中,类似于背包中的依赖物品的关系(参见背包九讲)。
此外,从状态考虑上来看,对于每个子树的根节点u都应该包含可以描述u的子树v状态的方程,因此设置状态方程为。
d
[
u
]
[
i
]
[
j
]
=
m
a
x
(
d
p
[
u
]
[
i
−
1
]
[
j
−
k
]
+
d
p
[
v
]
[
s
v
]
[
k
]
)
d[u][i][j] = max(dp[u][i-1][j-k] + dp[v][s_v][k])
d[u][i][j]=max(dp[u][i−1][j−k]+dp[v][sv][k])
表示以u为根节点,选择前i个子树,选了j门课程的最大学分,那么转移方程表示的是考虑第i个子树,选择第i个子树的中的k个课程的最大学分。
状态转移方程可以使用滚动数组实现,节省空间。即把第二维优化掉。因为当遍历第i个子树的时候一定是在1到i-1这些子树已经处理好的状态中继续计算的,所以第i个子树利用滚动数组的方式可以实现空间优化。
树形背包的入门题目,好好揣摩一下