思路是贪心加01背包
题目描述
岩石怪物杜达生活在魔法森林中,他在午餐时收集了 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和i+1位置交换,得到i+1和i
首先来看i和i+1的性质
再来看看i+1和i的性质
假设i和i+1是最优解的话就会使得
整理可得
得到这个公式后将结构体按照这个排序再进行01背包。
分析01背包其中f[i][j]表示的是所有以前i个能量石中选总时间恰好为j的最大能量
就可以得到状态转移方程为f[i][j]=max(f[i-1][j],f[i-1][j-s]+e-(j-s)*l
完整代码
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=11000;
int f[N];
struct Rec{
int s,e,l;
bool operator <(const Rec&t)const //从题目中可以得到一个性质就是
//S(i+1)*L(i)>S(i)*L(i+1),然后重载一下小于号用于结构体排序
{
return t.l*s<l*t.s;
}
}a[N];//表示能量石
int n;
int main()
{
int T;
cin>>T;//几组数据
for(int i=1;i<=T;i++)
{
int m=0; //表示如果将所有的能量石都消耗完后所需要的时间
scanf("%d",&n); //能量石的个数
for(int i=0;i<n;i++)
{
int s,l,e; //输入能量石的属性
scanf("%d%d%d",&s,&e,&l);
a[i]={s,e,l};
m+=s;//记录时间
}
sort(a,a+n);//然后将结构体按照S(i)*L(i+1)<S(i+1)*L(i)进行排序
memset(f,-0x3f,sizeof(f)); //因为是恰好时间为m的时候所获取的最大能量,所以将不满足条件的设为负无穷
f[0]=0;//当时间为零的时候所获取的能量为零
//然后再进行01背包
for(int j=0;j<n;j++)
{
int s=a[j].s,e=a[j].e,l=a[j].l;
for(int k=m;k>=s;k--)//
{
f[k]=max(f[k],f[k-s]+e-(k-s)*l);
}
}
//找到最大值
int res=0;
for(int j=0;j<=m;j++)res=max(res,f[j]);
printf("Case #%d: %d\n",i,res);
}
return 0;
}