【LeetCode 871】最低加油次数 动态规划DP 详解

汽车从起点出发驶向目的地,该目的地位于出发位置东面 target 英里处。

沿途有加油站,每个 station[i] 代表一个加油站,它位于出发位置东面 station[i][0] 英里处,并且有 station[i][1] 升汽油。

假设汽车油箱的容量是无限的,其中最初有 startFuel 升燃料。它每行驶 1 英里就会用掉 1 升汽油。

当汽车到达加油站时,它可能停下来加油,将所有汽油从加油站转移到汽车中。

为了到达目的地,汽车所必要的最低加油次数是多少?如果无法到达目的地,则返回 -1 。

注意:如果汽车到达加油站时剩余燃料为 0,它仍然可以在那里加油。如果汽车到达目的地时剩余燃料为 0,仍然认为它已经到达目的地。

示例 1:

输入:target = 1, startFuel = 1, stations = []
输出:0
解释:我们可以在不加油的情况下到达目的地。
示例 2:

输入:target = 100, startFuel = 1, stations = [[10,100]]
输出:-1
解释:我们无法抵达目的地,甚至无法到达第一个加油站。
示例 3:

输入:target = 100, startFuel = 10, stations = [[10,60],[20,30],[30,30],[60,40]]
输出:2
解释:
我们出发时有 10 升燃料。
我们开车来到距起点 10 英里处的加油站,消耗 10 升燃料。将汽油从 0 升加到 60 升。
然后,我们从 10 英里处的加油站开到 60 英里处的加油站(消耗 50 升燃料),
并将汽油从 10 升加到 50 升。然后我们开车抵达目的地。
我们沿途在1两个加油站停靠,所以返回 2 。

提示:

1 <= target, startFuel, stations[i][1] <= 10^9
0 <= stations.length <= 500
0 < stations[0][0] < stations[1][0] < … < stations[stations.length-1][0] < target

来源:力扣(LeetCode)

题意:如题

思路:

标程似乎是贪心+优先队列。我这里用比较慢但是很朴素的DP法过了。
用dp[i][j]表示前i个加油站,加油j次,能达到的最远距离。这样,我们最后只需要考虑dp[n][i]中最小能达到target的i即是答案。
状态转移:
外层遍历1->n个站台,内层遍历加j次油。那么对于当前j:
1.若前面i-1个站加j次油能到我i站台的位置,那么当前第一步我可以看成dp[i][j] = dp[i-1][j],先考虑从前i-1个站台转移过来,表示我用前i-1个站台加j次油的动力到达了i。
2.前面是前面的站加满j次,本站不加油。那么还有一种情况就是前面i-1个站只加到j-1次油,第j次在本站加,那么这一次便是dp[i][j] = dp[i-1][j-1] + 本站油量。
所以,对于这两个状态,挑一个最大的转移过来即可。 即是转台转移方程为(i从1开始):

dp[i][j] = max(dp[i-1][j],dp[i-1][j-1] + stations[i-1][1])

AC代码:

class Solution {
public:
    typedef long long ll;
    #define rep(i,a,n) for(int i=a;i<=n;++i)
    #define per(i,n,a) for(int i=n;i>=a;--i)
    #define pb push_back
    ll dp[505][505]; //dp[i][j]表示到前i个站加油j次所能到达最远距离
    const ll inf = 0x3f3f3f3f3f3f3f3f;
    int minRefuelStops(int target, int startFuel, vector<vector<int>>& stations) {
        int n = stations.size(); memset(dp,0,sizeof(dp));
        if(n==0) 
        {
            if(startFuel < target) return -1;
            else return 0;  
        }
        if(startFuel < stations[0][0]) return -1;
        dp[0][0] = startFuel;
        rep(i,1,n) rep(j,0,i)
        {
            if(dp[i-1][j] >= stations[i-1][0]) dp[i][j] = dp[i-1][j];
            if(j&&dp[i-1][j-1]>= stations[i-1][0] )
            dp[i][j] = max(dp[i][j],dp[i-1][j-1] + stations[i-1][1]);
        }
        rep(i,0,n) 
        {
            if(dp[n][i] >= target) return i;
        }
        return -1;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_45492531/article/details/107698365
今日推荐