LeetCode_差分数组_中等_1094.拼车

x33g5p2x  于2022-02-12 转载在 其他  
字(1.2k)|赞(0)|评价(0)|浏览(253)

1.题目

假设你是一位顺风车司机,车上最初有 capacity 个空座位可以用来载客。由于道路的限制,车只能向一个方向行驶(也就是说,不允许掉头或改变方向,你可以将其想象为一个向量)。
这儿有一份乘客行程计划表 trips[][],其中 trips[i] = [num_passengers, start_location, end_location] 包含了第 i 组乘客的行程信息:[ 必须接送的乘客数量;乘客的上车地点;以及乘客的下车地点 ]
这些给出的地点位置是从你的初始出发位置向前行驶到这些地点所需的距离(它们一定在你的行驶方向上)。

请你根据给出的行程计划表和车子的座位数,来判断你的车是否可以顺利完成接送所有乘客的任务(当且仅当你可以在所有给定的行程中接送所有乘客时,返回 true,否则请返回 false)

示例 1:
输入:trips = [[2,1,5],[3,3,7]], capacity = 4
输出:false

示例 2:
输入:trips = [[2,1,5],[3,3,7]], capacity = 5
输出:true

示例 3:
输入:trips = [[2,1,5],[3,5,7]], capacity = 3
输出:true

示例 4:
输入:trips = [[3,2,7],[3,7,9],[8,3,9]], capacity = 11
输出:true

提示:
你可以假设乘客会自觉遵守 “先下后上” 的良好素质
trips.length <= 1000
trips[i].length == 3
1 <= trips[i][0] <= 100
0 <= trips[i][1] < trips[i][2] <= 1000
1 <= capacity <= 100000

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/car-pooling

2.思路

(1)差分数组
参考小而美的算法技巧:差分数组

3.代码实现(Java)

//思路1————差分数组
public boolean carPooling(int[][] trips, int capacity) {
   //stations[i]:第i个车站的乘客数量(0≤i≤1000)
    int[] stations = new int[1001];
    for (int[] trip : trips) {
        /*
            trip[0]:乘客数量
            trip[1]:起始站
            trip[2]:终点站
        */
        //乘客上车,加上其数量
        stations[trip[1]] += trip[0];
        //乘客下车,减去其数量
        stations[trip[2]] -= trip[0];
    }
    //从起始站开始计数,如果乘客数量大于capacity,则认为超载
    int passengers = 0;
    for (int i : stations) {
        passengers += i;
        if (passengers > capacity) {
            return false;
        }
    }
    return true;
}

相关文章

微信公众号

最新文章

更多