2008. 出租车的最大盈利
中等
你驾驶出租车行驶在一条有 n
个地点的路上。这 n
个地点从近到远编号为 1
到 n
,你想要从 1
开到 n
,通过接乘客订单盈利。你只能沿着编号递增的方向前进,不能改变方向。
乘客信息用一个下标从 0 开始的二维数组 rides
表示,其中 rides[i] = [starti, endi, tipi]
表示第 i
位乘客需要从地点 starti
前往 endi
,愿意支付 tipi
元的小费。
每一位 你选择接单的乘客 i
,你可以 盈利 endi - starti + tipi
元。你同时 最多 只能接一个订单。
给你 n
和 rides
,请你返回在最优接单方案下,你能盈利 最多 多少元。
**注意:**你可以在一个地点放下一位乘客,并在同一个地点接上另一位乘客。
示例 1:
输入:n = 5, rides = [[2,5,4],[1,5,1]]
输出:7
解释:我们可以接乘客 0 的订单,获得 5 - 2 + 4 = 7 元。
示例 2:
输入:n = 20, rides = [[1,6,1],[3,10,2],[10,12,3],[11,12,2],[12,15,2],[13,18,1]]
输出:20
解释:我们可以接以下乘客的订单:
- 将乘客 1 从地点 3 送往地点 10 ,获得 10 - 3 + 2 = 9 元。
- 将乘客 2 从地点 10 送往地点 12 ,获得 12 - 10 + 3 = 5 元。
- 将乘客 5 从地点 13 送往地点 18 ,获得 18 - 13 + 1 = 6 元。
我们总共获得 9 + 5 + 6 = 20 元。
提示:
1 <= n <= 105
1 <= rides.length <= 3 * 104
rides[i].length == 3
1 <= starti < endi <= n
1 <= tipi <= 105
动态规划(选还是不选)
class Solution {
public long maxTaxiEarnings(int n, int[][] rides) {
// 按照出发时间排序
Arrays.sort(rides, (a, b) -> a[0] == b[0] ? a[1] - b[1] : a[0] - b[0]);
long[] f = new long[2 * n];
int idx = 0, m = rides.length;
// 从小到大枚举每个地点
for(int i = 0; i <= n; i++){
while(idx < m && rides[idx][0] < i)
idx++;
// 不选
f[i+1] = Math.max(f[i], f[i+1]);
// 当有乘客在地点i处出发时,可以选择载
// 那么就是 f[end+1] = f[i+1] + (end-start+tip)
while(idx < m && rides[idx][0] == i){
int start = rides[idx][0], end = rides[idx][1], tip = rides[idx][2];
f[end+1] = Math.max(f[end+1], f[i+1] + (end - start + tip));
idx++; // 可能会有多个乘客在i地点触发
}
}
return f[n+1];
}
}
记忆化搜索 ==> 动态规划
https://leetcode.cn/problems/maximum-earnings-from-taxi/solutions/2558504/jiao-ni-yi-bu-bu-si-kao-dong-tai-gui-hua-k15a/?envType=daily-question&envId=2023-12-08
class Solution {
public long maxTaxiEarnings(int n, int[][] rides) {
// 对rides按照end分组,对相同的end,记录start以及end-start+tip
List<int[]>[] groups = new ArrayList[n+1];
for(int[] r : rides){
int start = r[0], end = r[1], tip = r[2];
if(groups[end] == null)
groups[end] = new ArrayList<>();
groups[end].add(new int[]{start, end - start + tip});
}
long[] cache = new long[n+1];
Arrays.fill(cache, -1);
return dfs(n, groups, cache);
}
// 定义dfs(i)表示从1开到i最多可以赚多少钱
// 转移 枚举 i处下车 的乘客,取其中最赚钱的方案
// 不载 dfs(i) = dfs(i-1)
// 载 dfs(i) = max(dfs(start) + i - start + tip),其中满足 i=end
// 取所有情况最大值
// 递归边界:dfs(1)=0
// 递归入口 dfs(n)
public long dfs(int i, List<int[]>[] groups, long[] cache){
if(i == 1)
return 0;
if(cache[i] >= 0) return cache[i];
long res = dfs(i-1, groups, cache);
if(groups[i] != null){
for(int[] p : groups[i])
res = Math.max(res, dfs(p[0], groups, cache) + p[1]);
}
return cache[i] = res;
}
}
转为递推
class Solution {
public long maxTaxiEarnings(int n, int[][] rides) {
// 对rides按照end分组,对相同的end,记录start以及end-start+tip
List<int[]>[] groups = new ArrayList[n+1];
for(int[] r : rides){
int start = r[0], end = r[1], tip = r[2];
if(groups[end] == null)
groups[end] = new ArrayList<>();
groups[end].add(new int[]{start, end - start + tip});
}
long[] f = new long[n+1];
for(int i = 2; i <= n; i++){
f[i] = f[i-1];
if(groups[i] != null){
for(int[] p : groups[i]){
f[i] = Math.max(f[i], f[p[0]] + p[1]);
}
}
}
return f[n];
}
}