题目说明:在一条路上有N个加油站,给出每个加油站的价格和距离,计算到目的地的最小花费。油箱有最大容量,若无法到达目的地,计算出最大能走的距离。
思路:贪心算法。
首先按照距离升序排序。
遍历一次,判断能否走到目的地。这里需要注意第一个站距离大于0的情况。
再遍历一次,计算花费。对第i个站,判断i+1站是否更便宜,若是,加到刚好能到i+1站;否则,找到下一个更便宜的站j,判断油量是否足够到达,若足够,加到刚好能到j站;否则,加满油箱,并记下下一个需要加油的站和到这个站的剩余油量。然后对下一个需要加油的站进行判断。
代码:
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>
using namespace std;
float cmax,D,avg;
int N;
struct Station {
float price;
float dist;
bool operator<(const Station &a)const{
return dist<a.dist;
}
};
vector<Station> sta;
int main(int argc, const char * argv[]) {
cin>>cmax>>D>>avg>>N;
for (int i=0; i<N; i++) {
Station s;
cin>>s.price>>s.dist;
sta.push_back(s);
}
//把目的地加入向量,价格为0
Station s;
s.price=0;
s.dist=D;
sta.push_back(s);
//按照距离升序排序
sort(sta.begin(), sta.end());
float curgas=0;
float curprice=0;
//判断是否能到达目的地
//case 2:第一个站的距离大于0
if(sta[0].dist>0){
printf("The maximum travel distance = 0.00");
return 0;
}
for (int i=0; i<N; i++) {
if((sta[i+1].dist-sta[i].dist)/avg>cmax){
printf("The maximum travel distance = %.2f",cmax*avg+sta[i].dist);
return 0;
}
}
for (int i=0; i<N; ) {
//若下一个站更便宜,加到恰好能到下一个站
if(sta[i].price>=sta[i+1].price){
curprice+=((sta[i+1].dist-sta[i].dist)/avg-curgas)*sta[i].price;
curgas=(sta[i+1].dist-sta[i].dist)/avg;
i++;
curgas=0;
}
else{
for (int j=i+1; j<N+1; j++) {
//找到下一个更便宜的站,j
if(sta[i].price>sta[j].price){
//不能加到下一个便宜站
if((sta[j].dist-sta[i].dist)/avg>cmax){
//加满油箱
curprice+=(cmax-curgas)*sta[i].price;
curgas=cmax;
//找到在哪一个站需要补充
for (int k=i+1; k<N+1; k++) {
if((sta[k].dist-sta[i].dist)/avg>cmax){
curgas=cmax-(sta[k-1].dist-sta[i].dist)/avg;
i=k-1;
break;
}
}
}
//可以加到下一个便宜站
else{
curprice+=((sta[j].dist-sta[i].dist)/avg-curgas)*sta[i].price;
curgas=(sta[j].dist-sta[i].dist)/avg;
i=j;
curgas=0;
}
break;
}
}
}
}
printf("%.2f",curprice);
return 0;
}