题目
给你一个 互不相同 的整数数组,其中 locations[i] 表示第 i 个城市的位置。同时给你 start,finish 和 fuel 分别表示出发城市、目的地城市和你初始拥有的汽油总量
每一步中,如果你在城市 i ,你可以选择任意一个城市 j ,满足 j != i 且 0 <= j < locations.length ,并移动到城市 j 。从城市 i 移动到 j 消耗的汽油量为 |locations[i] - locations[j]|,|x| 表示 x 的绝对值。
请注意, fuel 任何时刻都 不能 为负,且你 可以 经过任意城市超过一次(包括 start 和 finish )。
请你返回从 start 到 finish 所有可能路径的数目。
由于答案可能很大, 请将它对 10^9 + 7 取余后返回。
示例 1:
输入:locations = [2,3,6,8,4], start = 1, finish = 3, fuel = 5
输出:4
解释:以下为所有可能路径,每一条都用了 5 单位的汽油:
1 -> 3
1 -> 2 -> 3
1 -> 4 -> 3
1 -> 4 -> 2 -> 3
示例 2:
输入:locations = [4,3,1], start = 1, finish = 0, fuel = 6
输出:5
解释:以下为所有可能的路径:
1 -> 0,使用汽油量为 fuel = 1
1 -> 2 -> 0,使用汽油量为 fuel = 5
1 -> 2 -> 1 -> 0,使用汽油量为 fuel = 5
1 -> 0 -> 1 -> 0,使用汽油量为 fuel = 3
1 -> 0 -> 1 -> 0 -> 1 -> 0,使用汽油量为 fuel = 5
示例 3:
输入:locations = [5,2,1], start = 0, finish = 2, fuel = 3
输出:0
解释:没有办法只用 3 单位的汽油从 0 到达 2 。因为最短路径需要 4 单位的汽油。
提示:
2 <= locations.length <= 100
1 <= locations[i] <= 109
所有 locations 中的整数 互不相同 。
0 <= start, finish < locations.length
1 <= fuel <= 200
思路
两点之间,直线距离最短。所以从点A到点B,如果不能一步直接到达,则也不能经过第三点到达。
所以如果点A到点B需要的fuel大于当前拥有的fuel,则从点A到达点B的路径数为0。
定义dp[i][j]表示拥有j的汽油时,从点i到终点的路径数。则答案为dp[start][fuel]。
dp[i][j] = dp[i][j]+dp[k][j-fuelk],0<=k<locations.size()。
fuelk为点i到点k消耗的汽油。
如果start和finish相等,认为可以直接到达,路径数加1。
代码
class Solution {
vector<vector<int>>dp;
int mod = 1000000007;
public:
int countRoutes(vector<int>& locations, int start, int finish, int fuel) {
int n = locations.size();
dp.assign(n,vector<int>(fuel+1,-1));
return routes(locations,start,finish,fuel);
}
int routes(vector<int>& locations, int start, int finish, int fuel) {
if(dp[start][fuel] != -1) {
return dp[start][fuel];
}
dp[start][fuel] = 0;
if(abs(locations[start]-locations[finish]) > fuel) {
return 0;
}
for(int i = 0;i < locations.size();i++) {
if (i != start) {
int need = abs(locations[i] - locations[start]);
if (need <= fuel) {
dp[start][fuel] += routes(locations,i,finish,fuel-need);
dp[start][fuel] %= mod;
}
}
}
if (start == finish) {
dp[start][fuel] += 1;
dp[start][fuel] %= mod;
}
return dp[start][fuel];
}
};