宣传一下算法提高课整理 <---
CSDN个人主页:更好的阅读体验 <---
本题链接(AcWing) 点这里
题目描述
岩石怪物杜达生活在魔法森林中,他在午餐时收集了 N N N 块能量石准备开吃。
由于他的嘴很小,所以一次只能吃一块能量石。
能量石很硬,吃完需要花不少时间。
吃完第 i i i 块能量石需要花费的时间为 S i S_i Si 秒。
杜达靠吃能量石来获取能量。
不同的能量石包含的能量可能不同。
此外,能量石会随着时间流逝逐渐失去能量。
第 i i i 块能量石最初包含 E i E_i Ei 单位的能量,并且每秒将失去 L i L_i Li 单位的能量。
当杜达开始吃一块能量石时,他就会立即获得该能量石所含的全部能量(无论实际吃完该石头需要多少时间)。
能量石中包含的能量最多降低至 0 0 0。
请问杜达通过吃能量石可以获得的最大能量是多少?
输入格式
第一行包含整数 T T T,表示共有 T T T 组测试数据。
每组数据第一行包含整数 N N N,表示能量石的数量。
接下来 N N N 行,每行包含三个整数 S i , E i , L i S_i,E_i,L_i Si,Ei,Li。
输出格式
每组数据输出一个结果,每个结果占一行。
结果表示为 Case #x: y
,其中 x x x 是组别编号(从 1 1 1 开始), y y y 是可以获得的最大能量值。
数据范围
1 ≤ T ≤ 10 1 \le T \le 10 1≤T≤10,
1 ≤ N ≤ 100 1 \le N \le 100 1≤N≤100,
1 ≤ S i ≤ 100 1 \le S_i \le 100 1≤Si≤100,
1 ≤ E i ≤ 1 0 5 1 \le E_i \le 10^5 1≤Ei≤105,
0 ≤ L i ≤ 1 0 5 0 \le L_i \le 10^5 0≤Li≤105
输入样例:
3
4
20 10 1
5 30 5
100 30 1
5 80 60
3
10 4 1000
10 3 1000
10 8 1000
2
12 300 50
5 200 0
输出样例:
Case #1: 105
Case #2: 8
Case #3: 500
样例解释
在样例#1中,有 N = 4 N = 4 N=4 个宝石。杜达可以选择的一个吃石头顺序是:
- 吃第四块石头。这需要 5 5 5 秒,并给他 80 80 80 单位的能量。
- 吃第二块石头。这需要 5 5 5 秒,并给他 5 5 5 单位的能量(第二块石头开始时具有 30 30 30 单位能量, 5 5 5 秒后失去了 25 25 25 单位的能量)。
- 吃第三块石头。这需要 100 100 100 秒,并给他 20 20 20 单位的能量(第三块石头开始时具有 30 30 30 单位能量, 10 10 10 秒后失去了 10 10 10 单位的能量)。
- 吃第一块石头。这需要 20 20 20 秒,并给他 0 0 0 单位的能量(第一块石头以 10 10 10 单位能量开始, 110 110 110 秒后已经失去了所有的能量)。
他一共获得了 105 105 105 单位的能量,这是能获得的最大值,所以答案是 105 105 105。
在样本案例#2中,有 N = 3 N = 3 N=3 个宝石。
无论杜达选择吃哪块石头,剩下的两个石头的能量都会耗光。
所以他应该吃第三块石头,给他提供 8 8 8 单位的能量。
在样本案例#3中,有 N = 2 N = 2 N=2 个宝石。杜达可以:
- 吃第一块石头。这需要 12 12 12 秒,并给他 300 300 300 单位的能量。
- 吃第二块石头。这需要 5 5 5 秒,并给他 200 200 200 单位的能量(第二块石头随着时间的推移不会失去任何能量!)。
所以答案是 500 500 500。
思路
本题为DP问题,可以使用 闫氏DP分析法 解题。
对于这种体积和价值变化的背包问题,我们考虑贪心。
对于两个相邻元素 i , j i,j i,j 的排列 ( i , j i,j i,j 获得的能量都不为 0 0 0)
− , − , − , i , j , − , − , − -,-,-,i,j,-,-,- −,−,−,i,j,−,−,−
显然有
E j + E i − S j × L i ≤ E i + E j − S i × L j E_j+E_i-S_j \times L_i \le E_i+E_j-S_i \times L_j Ej+Ei−Sj×Li≤Ei+Ej−Si×Lj
即
S j × L i ≥ S i × L j S_j \times L_i \ge S_i \times L_j Sj×Li≥Si×Lj
所以我们按照这个顺序将原序列排序,即可获得最优解的序列。
DP:
- 状态表示 f i , j f_{i,j} fi,j:
- 集合:考虑前 i i i 个魔法石,并且在吃掉某个魔法石后,总时间恰好为 j j j 的方案。
- 属性: max \max max
- 状态计算:
- 选第 i i i 个魔法石: f i − 1 , j − s i + e i − l i × ( j − s i ) f_{i-1,j-s_i}+e_i-l_i \times (j-s_i) fi−1,j−si+ei−li×(j−si)
- 不选第 i i i 个魔法石: f i − 1 , j f_{i-1,j} fi−1,j
时间复杂度
O ( n × ∑ i = 1 n s i ) O(n \times \sum_{i=1}^n s_i) O(n×i=1∑nsi)
AC Code
:
C + + C++ C++
cpp
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 10010;
int n;
struct ES
{
int s, e, l;
}st[N];
int f[N];
bool cmp(ES a, ES b) // 按s[i] * l[j] < s[j] * l[i]排序
{
return a.s * b.l < a.l * b.s;
}
int main()
{
int t;
cin >> t;
for (int q = 1; q <= t; q ++ )
{
cin >> n;
int m = 0;
for (int i = 0; i < n; i ++ )
{
int s, e, l;
cin >> s >> e >> l;
st[i] = {s, e, l};
m += s;
}
sort(st, st + n, cmp);
memset(f, -0x3f, sizeof f); // 对于"恰好"之类的问题,先把整个初始化成无穷大
f[0] = 0; // 起点为0
for (int i = 0; i < n; i ++ )
{
int s = st[i].s, e = st[i].e, l = st[i].l;
for (int j = m; j >= s; j -- )
f[j] = max(f[j], f[j - s] + e - (j - s) * l);
}
int res = 0;
for (int i = 1; i <= m; i ++ )
res = max(res, f[i]);
printf("Case #%d: %d\n", q, res); // 注意输出格式
}
return 0;
}
最后,如果觉得对您有帮助的话,点个赞再走吧!