【ACWing】1022. 宠物小精灵之收服

题目地址：

https://www.acwing.com/problem/content/1024/

宠物小精灵是一部讲述小智和他的搭档皮卡丘一起冒险的故事。一天，小智和皮卡丘来到了小精灵狩猎场，里面有很多珍贵的野生宠物小精灵。小智也想收服其中的一些小精灵。然而，野生的小精灵并不那么容易被收服。对于每一个野生小精灵而言，小智可能需要使用很多个精灵球才能收服它，而在收服过程中，野生小精灵也会对皮卡丘造成一定的伤害（从而减少皮卡丘的体力）。当皮卡丘的体力小于等于$0$时，小智就必须结束狩猎（因为他需要给皮卡丘疗伤），而使得皮卡丘体力小于等于$0$的野生小精灵也不会被小智收服。当小智的精灵球用完时，狩猎也宣告结束。我们假设小智遇到野生小精灵时有两个选择：收服它，或者离开它。如果小智选择了收服，那么一定会扔出能够收服该小精灵的精灵球，而皮卡丘也一定会受到相应的伤害；如果选择离开它，那么小智不会损失精灵球，皮卡丘也不会损失体力。小智的目标有两个：主要目标是收服尽可能多的野生小精灵；如果可以收服的小精灵数量一样，小智希望皮卡丘受到的伤害越小（剩余体力越大），因为他们还要继续冒险。现在已知小智的精灵球数量和皮卡丘的初始体力，已知每一个小精灵需要的用于收服的精灵球数目和它在被收服过程中会对皮卡丘造成的伤害数目。请问，小智该如何选择收服哪些小精灵以达到他的目标呢？

输入格式：
输入数据的第一行包含三个整数：$N$，$M$，$K$，分别代表小智的精灵球数量、皮卡丘初始的体力值、野生小精灵的数量。之后的$K$行，每一行代表一个野生小精灵，包括两个整数：收服该小精灵需要的精灵球的数量，以及收服过程中对皮卡丘造成的伤害。

输出格式：
输出为一行，包含两个整数：$C$，$R$，分别表示最多收服$C$个小精灵，以及收服$C$个小精灵时皮卡丘的剩余体力值最多为$R$。

数据范围：
$0<N\le 1000$
$0<M\le 500$
$0<K\le 100$

这是个二维费用的$0-1$背包问题，可以用动态规划来做。设$f[i][j][l]$是如果只收服前$i$个野生小精灵，并且使用不超过$j$个精灵球、皮卡丘消耗的体力也小于等于$l$的情况下，能收服的最多的小精灵数。那么可以按照第$i$个野生小精灵是否收服来分类，如果不收服，则答案是$f[i-1][j][l]$；如果收服，则答案是$f[i-1][j-v_1[i]][l-v_2[i]]$，其中$v_1[i]$和$v_2[i]$分别表示该野生小精灵消耗的精灵球数和皮卡丘体力。代码如下：

#include <iostream>
using namespace std;

const int N = 1010, M = 510, K = 110;
int n, m, k;
// v1是消耗精灵球数量，v2是消耗皮卡丘的体力
int v1[K], v2[K];
// 可以滚动数组优化空间
int f[2][N][M];

int main() {
    // n是精灵球数量，m是皮卡丘体力值，k是野生小精灵数量
    cin >> n >> m >> k;
    for (int i = 1; i <= k; i++) scanf("%d%d", &v1[i], &v2[i]);

    for (int i = 1; i <= k; i++)
        for (int j = 0; j <= n; j++)
        	// 消耗不能超过m - 1，因为最后皮卡丘的体力不能为0
            for (int l = 0; l < m; l++) {
            	// 枚举不收服第i个小精灵的情形
                f[i & 1][j][l] = f[i - 1 & 1][j][l];
                // 枚举收服第i个小精灵的情形
                if (j >= v1[i] && l >= v2[i])
                    f[i & 1][j][l] = max(f[i & 1][j][l], f[i - 1 & 1][j - v1[i]][l - v2[i]] + 1);
            }
    
    int t = m - 1;
    while (t > 0 && f[k & 1][n][t - 1] == f[k & 1][n][m - 1]) 
    	t--;

    printf("%d %d\n", f[k & 1][n][m - 1], m - t);

    return 0;
}

时间复杂度$O(NMK)$，空间$O(NM)$。