其他分享
首页 > 其他分享> > LeetCode/最低加油次数

LeetCode/最低加油次数

作者:互联网

汽车从起点出发驶向目的地,该目的地位于出发位置东面 target 英里处。
沿途有加油站,每个 station[i] 代表一个加油站,它位于出发位置东面 station[i][0] 英里处,并且有 station[i][1] 升汽油。
假设汽车油箱的容量是无限的,其中最初有 startFuel 升燃料。它每行驶 1 英里就会用掉 1 升汽油。
当汽车到达加油站时,它可能停下来加油,将所有汽油从加油站转移到汽车中。
为了到达目的地,汽车所必要的最低加油次数是多少?如果无法到达目的地,则返回 -1

1. 贪心算法

类似青蛙跳跃,更新维护最远加油站,不足以到达下一加油站时选取已经过加油站的最大油量

class Solution {
public:
    int minRefuelStops(int target, int startFuel, vector<vector<int>>& stations) {
        priority_queue<int> q;
        stations.push_back({target, 0});//终点当做加油站
        int last = 0, ans = 0;
        for (int i = 0; i < stations.size(); ++i) {
            auto need = stations[i][0] - last;//需要的油量
            while (!q.empty() && startFuel < need) {
                startFuel += q.top(); q.pop(); //取已走过加油站最大油量
                ans++;//计数增加
            }
            if (startFuel >= need) {//油量满足需求
                startFuel -= need;//到达下一加油站
                q.push(stations[i][1]);//把油量加入优先队列
            } 
            else return -1;
            last = stations[i][0];//更新上一位置
        }
        return ans;//遍历结束说明能到达最后加油站,即终点
    }
};

2. 动态规划(哈希表)

记录到达i加油站时的状态,即对应加油次数和所剩油量
后面的状态只需从前一个状态转移记录,当存在相同加油次数的路径时,取较大剩余油量
最后记录所有能达到最终目标的次数,更新取最小值
这里存在很多剪枝的细节,以及把起点和终点当加油站的处理方式,方便统一操作,减少判断

class Solution {
public:
    int minRefuelStops(int target, int startFuel, vector<vector<int>>& stations) {
        stations.push_back({target,0});//把终点当做加油站,统一处理
        int n = stations.size();
        vector<map<int,long>> dp(n+1);//表示到达i车站时对应加油次数和所剩油量,这里利用哈希表的自动排序
        dp[0][0] = startFuel;
        int prepos = 0;//记录上一个加油站的位置
        int mincount = INT_MAX;
        for(int i=1;i<=n;i++){//遍历得到所有加油站状态
            //if(dp[i-1].size()==0) return -1;//不能到达前一个加油站,返回-1
            for (auto& it : dp[i - 1]) {//从上一个加油站所有状态开始判断,从小到大
            if(it.first>=mincount) break;//剪枝
            if (it.second >= target - prepos) { mincount = min(it.first, mincount);break; }//剪枝
            //如果能到达下一个加油站
            int remain = it.second - (stations[i - 1][0] - prepos);//剩余油量
            if (remain >= 0) {
                //相同次数时,保留剩余油量多的
                if(remain>dp[i][it.first])
                    dp[i][it.first] = remain;//不加油
                if((long)remain + stations[i - 1][1]> dp[i][it.first + 1])
                    dp[i][it.first + 1] = (long)remain + stations[i - 1][1];//加油
            }
        }
        prepos = stations[i - 1][0];//记录上一次加油站位置
    }
    if(mincount!=INT_MAX) return mincount;
    return -1;
    }
};

3. 动态规划(一维)

动态维护加油i次时的最远距离,新增加油站时,对所有值从后往前进行更新
dp[i]表示加油i次能到达的最远距离
这里其实也是经过二维动态规划压缩后的结果

class Solution {
public:
    int minRefuelStops(int target, int startFuel, vector<vector<int>>& stations) {
        vector<unsigned int> dp(stations.size() + 1);//dp[i]表示加油i次能到达的最远距离
        dp[0] = startFuel;
        //从前往后遍历考察每个加油站
        for(int i = 0; i < stations.size(); i++)
            //新增加油站后,从后往前更新dp表,从后往前才不会对下一个造成影响
            for(int j = i; j >= 0; j--)
                if(dp[j] >= stations[i][0])//如果加油j次能到达i加油站
                    dp[j + 1] = max(dp[j + 1], dp[j] + stations[i][1]);//更新j+1次的最远距离

        for(int i = 0; i < dp.size(); i++)
            if(dp[i] >= target) return i;//返回一个最小加油次数
        return -1;
    }
};

标签:stations,int,次数,加油站,startFuel,加油,LeetCode,dp
来源: https://www.cnblogs.com/929code/p/16439134.html