题目:
逗志芃励志要成为强力党,所以他将身上所以的技能点都洗掉了重新学技能。现在我们可以了解到,每个技能都有一个前提技能,只有学完了前提技能才能学习当前的技能(有一个最根本的技能不需要前提技能)。学习每个技能要消耗一个技能点,然后可以获得这个技能的威力值。由于逗志芃要陪妹子,所以他希望你教他如何点技能使得威力值最大从而成为强力党。
分析:本题是一个有依赖的背包问题,可以视作分组背包+01背包+dfs的结合体。下面具体解释为什么是分组背包+01背包+dfs。
代码:
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 210;
int n,m;
int q[N];
int h[N],e[N],ne[N],idx;
//h 是邻接表头数组,e 是邻接表中存储的节点
//ne 是邻接表中的下一个节点,idx 是邻接表的索引。
int dp[N][N];//以i为根节点,总花费的点数不超过j的最大方案
bool st[N];//是否有前驱
int res = 0;//伤害的最大的值
void add(int x,int y)
{
e[idx] = y,ne[idx] = h[x],h[x] = idx++;
//其实是对以x为根的树(每个都只看作只有一层子节点),建立从右往左遍历,最后遍历到根节点的顺序。
}
void dfs(int u)
{
for(int i = h[u];i != -1; i = ne[i])//循环物品组
{
int son = e[i];
dfs(son);
for(int j = m - 1; j >= 0; j--)//循环剩余的点数
for(int k = 0; k <= j; k++)//循环决策,分给子树的体积视作组里的物品
//比如根为1,有子节点2 3,那么对2和3的所有子节点的可能的体积分配视作两组,那么这一层就看做一个分组背包问题。
dp[u][j] = max(dp[u][j], dp[u][j - k] + dp[son][k]);//给
}
//将当前的节点1
for(int i = m; i >= 1; i--) dp[u][i] = dp[u][i - 1] + q[u];
dp[u][0] = 0;
}
int main()
{
cin >> n >> m;
memset(h, -1 ,sizeof h);
for(int i = 1; i <= n; i++)//读取每个节点的值
cin>>q[i];
for(int i = 1; i < n; i++)//读取节点之间的关系
{
int x,y;
cin >> x >> y;
add(x,y);//x为y的父节点
st[y] = true;
}
int root;//寻找根节点
for(int i = 1; i <= n; i++)
if(st[i] == false){
root = i;
break;
}
dfs(root);
cout << dp[root][m] << endl;
return 0;
}
以下是add函数的解释:
e[ ]数组是存放技能的序号,ne[ ]数组是用来从某棵子树的最右子孩子一直向左遍历所有的子孩子,h[i]是序号为i的技能最右子孩子的编号。