打怪兽问题
作者：Grey
原文地址：
题目描述
题目链接: 牛客-打怪兽
开始时你的能力是0，你的目标是从0号怪兽开始，通过所有的怪兽。
如果你当前的能力，小于i号怪兽的能力，你必须付出money[i]的钱，贿赂这个怪兽，然后怪兽就会加入你，他的能力直接累加到你的能力上；
如果你当前的能力，大于等于 i 号怪兽的能力，你可以选择直接通过，你的能力并不会下降，你也可以选择贿赂这个怪兽，然后怪兽就会加入你，他的能力直接累加到你的能力上。
返回通过所有的怪兽，需要花的最小钱数。
主要思路
本题要根据数据量不同使用不同的动态规划解法
情况1：如果怪兽数量不大和血量都不大的情况，定义如下递归函数
long p(int[] hp, int[] money, int i, int j)
递归含义是：当前血量是 j，从第 i 号怪兽开始，到最后通过所有怪兽的最少钱数是多少？
代码和注释说明如下
public static long p(int[] hp, int[] money, int i, int j) {
// i 已经到最后了，不需要继续花钱打怪兽
if (i == hp.length) {
return 0;
}
// i号怪兽选择贿赂
long p = money[i] + p(hp, money, i + 1, j + hp[i]);
// 不选贿赂i号怪兽，这时候是有条件的，就是当前血量 j 需要大于当前怪兽值
if (j >= hp[i]) {
return Math.min(p, p(hp, money, i + 1, j));
}
return p;
}
通过上述递归函数，可以转换成动态规划，利用一个二维数组即可，代码如下
public static long func1Dp(int[] hp, int[] money) {
int sum = 0;
for (int a : hp) {
sum += a;
}
long[][] dp = new long[hp.length + 1][sum + 1];
for (int i = hp.length - 1; i >= 0; i--) {
for (int j = sum - hp[i]; j >= 0; j--) {
dp[i][j] = money[i] + dp[i + 1][j + hp[i]];
if (j >= hp[i]) {
dp[i][j] = Math.min(dp[i][j], dp[i + 1][j]);
}
}
}
return dp[0][0];
}
这个二维数组的规模就是上述递归函数 i 和 j 的变化范围相乘，其中 i 的变化范围是[0……怪兽数量之和]，j 的变化范围是[0……血量和]。所以适用于怪兽数量和血量都不大的场景。
情况2：如果怪兽能力比较大，但是钱数不大情况，我们需要调整递归函数，定义如下递归函数
long p2(int[] hp, int[] money, int index, int m)
递归含义是：从 0……index 号怪兽，花的钱，必须严格等于 m 的情况下，如果通过不了，返回-1；如果可以通过，返回能通过情况下的最大能力值。
递归方法实现如下
public static long p2(int[] hp, int[] money, int index, int m) {
// 从右往左填，base case
if (index == -1) {
return m == 0 ? 0L : -1L;
}
// 贿赂当前怪兽
long p1 = p2(hp, money, index - 1, m);
long ans = -1;
if (p1 != -1 && p1 >= hp[index]) {
// 贿赂是有效的，或者贿赂结果可以支持下一次的决策
ans = p1;
}
// 不贿赂当前怪兽
long p2 = p2(hp, money, index - 1, m - money[index]);
if (p2 != -1) {
// 可以走到最后，与贿赂得到的能力值pk下。
ans = Math.max(ans, p2 + hp[index]);
}
return ans;
}
同理，这个递归也可以改成动态规划，也是利用一个二维数组，
public static long func4(int[] d, int[] p) {
int sum = 0;
for (int num : p) {
sum += num;
}
// dp[i][j]含义：
// 能经过0～i的怪兽，且花钱为j（花钱的严格等于j）时的武力值最大是多少？
// 如果dp[i][j]==-1，表示经过0～i的怪兽，花钱为j是无法通过的，或者之前的钱怎么组合也得不到正好为j的钱数
int[][] dp = new int[d.length][sum + 1];
for (int i = 0; i < dp.length; i++) {
for (int j = 0; j <= sum; j++) {
dp[i][j] = -1;
}
}
// 经过0～i的怪兽，花钱数一定为p[0]，达到武力值d[0]的地步。其他第0行的状态一律是无效的
dp[0][p[0]] = d[0];
for (int i = 1; i < d.length; i++) {
for (int j = 0; j <= sum; j++) {
// 可能性一，为当前怪兽花钱
// 存在条件：
// j - p[i]要不越界，并且在钱数为j - p[i]时，要能通过0～i-1的怪兽，并且钱数组合是有效的。
if (j >= p[i] && dp[i - 1][j - p[i]] != -1) {
dp[i][j] = dp[i - 1][j - p[i]] + d[i];
}
// 可能性二，不为当前怪兽花钱
// 存在条件：
// 0~i-1怪兽在花钱为j的情况下，能保证通过当前i位置的怪兽
if (dp[i - 1][j] >= d[i]) {
// 两种可能性中，选武力值最大的
dp[i][j] = Math.max(dp[i][j], dp[i - 1][j]);
}
}
}
int ans = 0;
// dp表最后一行上，dp[N-1][j]代表：
// 能经过0～N-1的怪兽，且花钱为j（花钱的严格等于j）时的武力值最大是多少？
// 那么最后一行上，最左侧的不为-1的列数(j)，就是答案
for (int j = 0; j <= sum; j++) {
if (dp[d.length - 1][j] != -1) {
ans = j;
break;
}
}
return ans;
}
可以看到二维数组的 dp 规模是钱数 * 达到的最大能力。适用于怪兽能力比较大，但是钱数不大的情况。
完整代码如下(含对数器)
public class NowCoder_BeatMonster {
// i到最后通过所有怪兽的最少钱数是多少？
// 适合怪兽能力不大的情况
public static long func1(int[] hp, int[] money) {
return p(hp, money, 0, 0);
}
public static long p(int[] hp, int[] money, int i, int j) {
if (i == hp.length) {
return 0;
}
// 选择贿赂
long p = money[i] + p(hp, money, i + 1, j + hp[i]);
// 不选贿赂，有条件
if (j >= hp[i]) {
return Math.min(p, p(hp, money, i + 1, j));
}
return p;
}
public static long func1Dp(int[] hp, int[] money) {
int sum = 0;
for (int a : hp) {
sum += a;
}
long[][] dp = new long[hp.length + 1][sum + 1];
for (int i = hp.length - 1; i >= 0; i--) {
for (int j = sum - hp[i]; j >= 0; j--) {
dp[i][j] = money[i] + dp[i + 1][j + hp[i]];
if (j >= hp[i]) {
dp[i][j] = Math.min(dp[i][j], dp[i + 1][j]);
}
}
}
return dp[0][0];
}
public static long func2(int[] hp, int[] money) {
int sum = 0;
for (int a : money) {
sum += a;
}
int N = hp.length;
for (int i = 0; i < sum; i++) {
if (p2(hp, money, N - 1, i) != -1) {
return i;
}
}
return sum;
}
// 从0....index 怪兽，花的钱，必须严格==m
// 如果通过不了，返回-1
// 如果可以通过，返回能通过情况下的最大能力值
public static long p2(int[] hp, int[] money, int index, int m) {
if (index == -1) {
return m == 0 ? 0L : -1L;
}
// 贿赂当前怪兽
long p1 = p2(hp, money, index - 1, m);
long ans = -1;
if (p1 != -1 && p1 >= hp[index]) {
ans = p1;
}
long p2 = p2(hp, money, index - 1, m - money[index]);
if (p2 != -1) {
ans = Math.max(ans, p2 + hp[index]);
}
return ans;
}
public static long func4(int[] d, int[] p) {
int sum = 0;
for (int num : p) {
sum += num;
}
// dp[i][j]含义：
// 能经过0～i的怪兽，且花钱为j（花钱的严格等于j）时的武力值最大是多少？
// 如果dp[i][j]==-1，表示经过0～i的怪兽，花钱为j是无法通过的，或者之前的钱怎么组合也得不到正好为j的钱数
int[][] dp = new int[d.length][sum + 1];
for (int i = 0; i < dp.length; i++) {
for (int j = 0; j <= sum; j++) {
dp[i][j] = -1;
}
}
// 经过0～i的怪兽，花钱数一定为p[0]，达到武力值d[0]的地步。其他第0行的状态一律是无效的
dp[0][p[0]] = d[0];
for (int i = 1; i < d.length; i++) {
for (int j = 0; j <= sum; j++) {
// 可能性一，为当前怪兽花钱
// 存在条件：
// j - p[i]要不越界，并且在钱数为j - p[i]时，要能通过0～i-1的怪兽，并且钱数组合是有效的。
if (j >= p[i] && dp[i - 1][j - p[i]] != -1) {
dp[i][j] = dp[i - 1][j - p[i]] + d[i];
}
// 可能性二，不为当前怪兽花钱
// 存在条件：
// 0~i-1怪兽在花钱为j的情况下，能保证通过当前i位置的怪兽
if (dp[i - 1][j] >= d[i]) {
// 两种可能性中，选武力值最大的
dp[i][j] = Math.max(dp[i][j], dp[i - 1][j]);
}
}
}
int ans = 0;
// dp表最后一行上，dp[N-1][j]代表：
// 能经过0～N-1的怪兽，且花钱为j（花钱的严格等于j）时的武力值最大是多少？
// 那么最后一行上，最左侧的不为-1的列数(j)，就是答案
for (int j = 0; j <= sum; j++) {
if (dp[d.length - 1][j] != -1) {
ans = j;
break;
}
}
return ans;
}
public static int[][] generateTwoRandomArray(int len, int value) {
int size = (int) (Math.random() * len) + 1;
int[][] arrs = new int[2][size];
for (int i = 0; i < size; i++) {
arrs[0][i] = (int) (Math.random() * value) + 1;
arrs[1][i] = (int) (Math.random() * value) + 1;
}
return arrs;
}
public static void main(String[] args) {
int len = 12;
int value = 20;
int testTimes = 10000;
for (int i = 0; i < testTimes; i++) {
int[][] arrs = generateTwoRandomArray(len, value);
int[] d = arrs[0];
int[] p = arrs[1];
long ans1 = func1(d, p);
long ans4 = func1Dp(d, p);
long ans2 = func2(d, p);
long ans3 = func4(d, p);
if (ans1 != ans2 || ans2 != ans3 || ans1 != ans4) {
System.out.println("oops!");
}
}
}
}
扫一扫
478
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
