PAT 1033. To Fill or Not to Fill (25)

题目说明：在一条路上有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;
}