【备战秋招】每日一题:塔子大厂真题模拟赛-第二题-魔法石(Ⅱ)

最新推荐文章于 2024-09-13 17:30:19 发布

塔子哥学算法

最新推荐文章于 2024-09-13 17:30:19 发布

阅读量78

点赞数

文章标签： java 开发语言算法深度优先

本文链接：https://blog.csdn.net/m0_73659489/article/details/131882968

版权

文章介绍了一个使用动态规划解决的问题，即在有限的能量下，如何分配以最大化魔法石点亮后的总价值。魔法石有初始、第一次点亮和第二次点亮后的三种不同价值，问题转换为分组背包问题。给出的解决方案是通过动态规划的状态转移实现，确保在不超过能量限制的情况下选取每组中的一项魔法石，以达到最大价值。代码示例分别用Python和C++展示了解决方法。

摘要由CSDN通过智能技术生成

为了更好的阅读体检，可以查看我的算法学习网
 本题在线评测链接:P1212

题目内容

塔子哥有 $n$ 个魔法石，初始时魔法石的价值为 $a_i$ .塔子哥还有 $m$ 点能量。对于每个能量，它可以用来点亮一颗魔法石。第 $i$ 个魔法石第一次被点亮后的价值为 $b_i$ ，魔法石也可以被再次点亮，价值进而变为 $c_i$ .请问塔子哥该如何安排手中的能量，使得魔法石的价值总和最大呢？

输入描述

第一行两个整数 $n$ , $m$ ( $\leq n \leq 1000 , 0 \leq m \leq 1000$ ) , 分别代表魔法石的个数以及能量数量.

第二行 $n$ 个整数: $a_1,a_2,...,a_n$ , 代表这 $n$ 个魔法石初始价值

第三行 $n$ 个整数: $b_1,b_2,...,b_n$ , 代表这 $n$ 个魔法石第一次被点亮后的价值

第三行 $n$ 个整数: $c_1,c_2,...,c_n$ , 代表这 $n$ 个魔法石第二次被点亮后的价值

$\leq a_i , b_i , c_i \leq 1e9$

输出描述

一行整数，代表最大价值。

样例1

输入

输出

样例2

输入

输出

思路:动态规划-分组背包

本问题等价于有一个容积为 $m$ 的背包，现有 $n$ 组物品，每组有三个物品，第 $i$ 组物品为:

物品1:价值 $a_i$ ,体积 $0$

物品2:价值 $b_i$ ,体积 $1$

物品3:价值 $c_i$ ,体积 $2$

要求：从这 $n$ 组物品中选出 $n$ 件物品(每组必选且只能选一件),要求容积不超过 $m$ 的最大价值。这是典型的分组背包。可以自行上网学习。推荐先搞懂01背包，再学习分组背包。细节见代码

时间复杂度

有 $nm$ 个状态，3次转移,总复杂度: $O (nm * 3) = O (nm) = 1 e 6 < 1 e 8$ ，可以1s过

代码

python

n , m = list(map(int , input().split()))
a = []
for i in range(3):
	a.append(list(map(int , input().split())))
# 类似01背包地，分组背包也能够优化空间 , dp[i] 代表当前大小恰好为i的体积的最优解
dp = [0] * (m + 1) 
# 考虑前n组物品
for i in range(n): 
    # <从大到小>枚举体积   -先枚举体积再枚举物品，保证组内物品互斥选择
	for j in range(m , -1 , -1):
        # 枚举组内物品
		g = dp[j]
		for k in range(2 , -1 , -1):
			if j - k >= 0:
                # 转移方程
				g = max(g , dp[j - k] + a[k][i])
		dp[j] = g
print(max(dp))

思考:如果本题 $a_i,b_i,c_i$ 允许为负数，上述代码是否依旧正确。如果不正确，将如何修改？

C++ (from 2333)

#include <bits/stdc++.h>
using namespace std;
void solve() {
	int n, m;
	cin >> n >> m;
	vector<int> a(n + 1), b(n + 1), c(n + 1);
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
	}
	for (int i = 1; i <= n; i++) {
		cin >> b[i];
	}
	for (int i = 1; i <= n; i++) {
		cin >> c[i];
	}
	vector<ll> dp(m + 1);
	for (int i = 1; i <= n; i++) {
		for (int j = m; j >= 0; j--) {
			dp[j] += a[i];
			if (j > 0) dp[j] = max(dp[j], dp[j - 1] + b[i]);
			if (j > 1) dp[j] = max(dp[j], dp[j - 2] + c[i]);
		}
	}
	cout << *max_element(dp.begin(), dp.end()) << endl;
}
int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    int t = 1;
    while (t--) {
    	solve();
	}
}