Google Kickstart2019 Round B Problem B Energy Stones 贪心+DP

题目描述

岩石怪物杜达生活在魔法森林中，他在午餐时收集了N块能量石准备开吃。
由于他的嘴很小，所以一次只能吃一块能量石。
能量石很硬，吃完需要花不少时间。
吃完第 i 块能量石需要花费的时间为Si秒。
杜达靠吃能量石来获取能量。
不同的能量石包含的能量可能不同。
此外，能量石会随着时间流逝逐渐失去能量。
第 i 块能量石最初包含Ei单位的能量，并且每秒将失去Li单位的能量。
当杜达开始吃一块能量石时，他就会立即获得该能量石所含的全部能量（无论实际吃完该石头需要多少时间）。
能量石中包含的能量最多降低至0。
请问杜达通过吃能量石可以获得的最大能量是多少？

输入格式
第一行包含整数T，表示共有T组测试数据。
每组数据第一行包含整数N，表示能量石的数量。
接下来N行，每行包含三个整数Si,Ei,Li。

输出格式
每组数据输出一个结果，每个结果占一行。
结果表示为“Case #x: y”，其中x是组别编号（从1开始），y是可以获得的最大能量值。

数据范围
1≤T≤10,
1≤N≤100,
1≤Si≤100,
1≤Ei≤105,
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个宝石。杜达可以选择的一个吃石头顺序是：

吃第四块石头。这需要5秒，并给他80单位的能量。
吃第二块石头。这需要5秒，并给他5单位的能量（第二块石头开始时具有30单位能量，5秒后失去了25单位的能量）。
吃第三块石头。这需要100秒，并给他20单位的能量（第三块石头开始时具有30单位能量，10秒后失去了10单位的能量）。
吃第一块石头。这需要20秒，并给他0单位的能量（第一块石头以10单位能量开始，110秒后已经失去了所有的能量）。
他一共获得了105单位的能量，这是能获得的最大值，所以答案是105。

在样本案例＃2中，有N = 3个宝石。

无论杜达选择吃哪块石头，剩下的两个石头的能量都会耗光。

所以他应该吃第三块石头，给他提供8单位的能量。

在样本案例＃3中，有N = 2个宝石。杜达可以：

吃第一块石头。这需要12秒，并给他300单位的能量。
吃第二块石头。这需要5秒，并给他200单位的能量（第二块石头随着时间的推移不会失去任何能量！）。
所以答案是500。

思路

• 我们将其排序， 考虑吃 i 和 i + 1 的情况， 先吃i Ei+Ei+1 - Si * Li+1 先吃 i + 1 得到 Ei+Ei+1 - Si + 1* Li 所以 我们得到， 我们想要的是 Si * Li+1 < Si + 1* Li, 我们按照这个顺序排列， 从前往后吃， 就可以得到最优解
• 求最优解时， 这是个01背包问题

代码

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>

using namespace std;

const int N = 10010;

int f[N];

struct node{
	int s, e, l;
	bool operator < (const node &a) const 
	{
		return a.l * s < l * a.s ;
	}
}stone[110];

int n, m;

int main()
{
	int t;
	cin >> t;
	int c = 0;
	while(t--)
	{
		cin >> n;
		m = 0;
		for(int i = 1; i <= n; i++)
		{
			int a, b, c;
			cin >> a >> b >> c;
			stone[i] = {a, b, c};
			m +=  a;
		}
		sort(stone + 1, stone + 1 + n);
		memset(f, -0x3f, sizeof f);
		f[0] = 0;
		for(int i = 1; i <= n; i++)
		{
			int s = stone[i].s;
			int e = stone[i].e;
			int l = stone[i].l;
			for(int j = m; j >= s; j--)
			{
				f[j] = max(f[j], f[j - s] + e - (j - s) * l);
			}
		}
		
		int ans = 0;
		for(int i = 1; i <= m; i++)
			ans = max(ans, f[i]);
		printf("Case #%d: %d\n", ++c, ans);
			
	}
	return 0;
}