acwing 1022. 宠物小精灵之收服 | 二维01背包
题目描述
宠物小精灵是一部讲述小智和他的搭档皮卡丘一起冒险的故事。
一天,小智和皮卡丘来到了小精灵狩猎场,里面有很多珍贵的野生宠物小精灵。
小智也想收服其中的一些小精灵。
然而,野生的小精灵并不那么容易被收服。
对于每一个野生小精灵而言,小智可能需要使用很多个精灵球才能收服它,而在收服过程中,野生小精灵也会对皮卡丘造成一定的伤害(从而减少皮卡丘的体力)。
当皮卡丘的体力小于等于0时,小智就必须结束狩猎(因为他需要给皮卡丘疗伤),而使得皮卡丘体力小于等于0的野生小精灵也不会被小智收服。
当小智的精灵球用完时,狩猎也宣告结束。
我们假设小智遇到野生小精灵时有两个选择:收服它,或者离开它。
如果小智选择了收服,那么一定会扔出能够收服该小精灵的精灵球,而皮卡丘也一定会受到相应的伤害;如果选择离开它,那么小智不会损失精灵球,皮卡丘也不会损失体力。
小智的目标有两个:主要目标是收服尽可能多的野生小精灵;如果可以收服的小精灵数量一样,小智希望皮卡丘受到的伤害越小(剩余体力越大),因为他们还要继续冒险。
现在已知小智的精灵球数量和皮卡丘的初始体力,已知每一个小精灵需要的用于收服的精灵球数目和它在被收服过程中会对皮卡丘造成的伤害数目。
请问,小智该如何选择收服哪些小精灵以达到他的目标呢?
输入格式
输入数据的第一行包含三个整数:N,M,K,分别代表小智的精灵球数量、皮卡丘初始的体力值、野生小精灵的数量。
之后的K行,每一行代表一个野生小精灵,包括两个整数:收服该小精灵需要的精灵球的数量,以及收服过程中对皮卡丘造成的伤害。
输出格式
输出为一行,包含两个整数:C,R,分别表示最多收服C个小精灵,以及收服C个小精灵时皮卡丘的剩余体力值最多为R。
数据范围
0<N≤10000<N≤1000,
0<M≤5000<M≤500,
0<K≤1000<K≤100输入样例1:
10 100 5 7 10 2 40 2 50 1 20 4 20
输出样例1:
3 30
输入样例2:
10 100 5 8 110 12 10 20 10 5 200 1 110
输出样例2:
0 100
二维01背包
分析
用\(f[i][j][z]\) 表示前\(i\)个小精灵,使用\(j\)个精灵球,体力耗费\(z\)的时候所能收服的最大精灵数量
-
当不选第i个小精灵 或者 精灵球数量/体力 不足以选第i个小精灵的时候:\(f[i][j][z] = f[i-1][j][z]\)
-
当可以选第i个小精灵的时候:\(f[i][j][z] = f[i-1][j - v1_i[z-v2_i] + 1\)
状态转移方程为:$f[i][j][z] =max(f[i-1][j][z], f[i-1][j - v1_i][z-v2_i] + 1) \quad if(j \ge v1_i \ and \ z \ge v2_i) $
与01背包类似优化成1维,然后此时能收服最大精灵数:f[n][m-1]
(因为收服之后体力不能为0)
for(int i = 1; i <= k; i++)
for(int j = n; j >= v1[i]; j--)
for(int z = m-1; z >= v2[i]; z--)
f[j][z] = max(f[j][z], f[j-v1[i]][z-v2[i]] + 1);
cout << f[n][m-1] << " ";
然后求出能够收服f[n][m-1]
个精灵所耗费的最小的体力r
, m-r
就是最大剩余体力
int r = m-1;
while(r > 0 && f[n][r-1] == f[n][m-1]) r--;
cout << m - r;
代码
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N = 1010;
int f[N][N];
int n, m, k;
int v1[N], v2[N];
int main()
{
cin >> n >> m >> k;
for(int i = 1; i <= k; i++) cin >> v1[i] >> v2[i];
/*
for(int i = 1; i <= k; i++) // 前i种
{
for(int j = 0; j <= n; j++) // 精灵球数量
{
for(int z = 0; z <= m-1; z++) // 血量,
{
f[i][j][z] = f[i-1][j][z];
if(j >= v1[i] && z >= v2[i])
f[i][j][z] = max(f[i][j][z], f[i-1][j-v1[i]][z-v2[i]] + 1)
}
}
} */
for(int i = 1; i <= k; i++)
for(int j = n; j >= v1[i]; j--)
for(int z = m-1; z >= v2[i]; z--)
f[j][z] = max(f[j][z], f[j-v1[i]][z-v2[i]] + 1);
cout << f[n][m-1] << " ";
int r = m-1;
while(r > 0 && f[n][r-1] == f[n][m-1]) r--;
cout << m - r;
return 0;
}