美文网首页
【教3妹学编程-算法题】拼车

【教3妹学编程-算法题】拼车

作者: 程序员小2 | 来源:发表于2023-12-01 15:45 被阅读0次
    阳光明媚

    3妹:“太阳当空照,花儿对我笑,小鸟说早早早,你为什么背上炸药包”
    2哥 :3妹,什么事呀这么开发。
    3妹:2哥你看今天的天气多好啊,阳光明媚、万里无云、秋高气爽,适合秋游。
    2哥:是啊,立冬之后天气多以多云为主,好不容易艳阳高照。可是你不能秋游,赶紧收拾收拾上班去啦
    3妹:哼, 好吧~
    2哥:给你出了一道题发你微信里了, 上班通勤的路上记得看一下,回来问你答案~

    image.png
    3妹:知道啦,难不倒我!

    题目:

    车上最初有 capacity 个空座位。车 只能 向一个方向行驶(也就是说,不允许掉头或改变方向)

    给定整数 capacity 和一个数组 trips , trip[i] = [numPassengersi, fromi, toi] 表示第 i 次旅行有 numPassengersi 乘客,接他们和放他们的位置分别是 fromi 和 toi 。这些位置是从汽车的初始位置向东的公里数。

    当且仅当你可以在所有给定的行程中接送所有乘客时,返回 true,否则请返回 false。

    示例 1:

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

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

    提示:

    1 <= trips.length <= 1000
    trips[i].length == 3
    1 <= numPassengersi <= 100
    0 <= fromi < toi <= 1000
    1 <= capacity <= 10^5

    思路:

    思考

    差分,
    从朴素的想法开始:创建一个数组 cnt,用于存储从某个站点出发时,车上的乘客数量。

    例如 cnt[x]=c含义为在站点 x 出发时(在该站点的下车和上车均完成),车上乘客数为 c 个。

    对于每个 trips[i]=(c,a,b),我们需要对 [a,b)范围内的 cnt[j]进行加 c 操作。

    java代码:

    class Solution {
        public boolean carPooling(int[][] trips, int capacity) {
            int toMax = 0;
            for (int[] trip : trips) {
                toMax = Math.max(toMax, trip[2]);
            }
    
            int[] diff = new int[toMax + 1];
            for (int[] trip : trips) {
                diff[trip[1]] += trip[0];
                diff[trip[2]] -= trip[0];
            }
    
            int count = 0;
            for (int i = 0; i <= toMax; ++i) {
                count += diff[i];
                if (count > capacity) {
                    return false;
                }
            }
            return true;
        }
    }
    

    相关文章

      网友评论

          本文标题:【教3妹学编程-算法题】拼车

          本文链接:https://www.haomeiwen.com/subject/zxmcgdtx.html