算法题:LeetCode (1094) 拼车【一题搞懂差分数组】

题目描述

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

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

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

示例

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



解题思路

差分数组:差分数组主要的适用场景是对原始数组进行频繁的区间增减操作,这个时候适用差分数组能够快速的完成,同时能够快速获得更新后的数组各个位置的值。

trips = [[2,1,5],[3,3,7]], capacity = 4为例,数组变化如下:
乘客情况

使用差分数组修改上图所示的数组,结果如下:
算法题:LeetCode (1094) 拼车【一题搞懂差分数组】_第1张图片



代码(Java)

public class LeetCode_1094_carPooling {
    public boolean carPooling(int[][] trips, int capacity) {
        int[] diff = new int[1001];
        for (int[] t : trips) {
            int passengers = t[0], from = t[1], to = t[2];
            diff[from] += passengers;
            diff[to] -= passengers;
        }
        int num = 0;    // 差分求和
        for (int i : diff) {
            num += i;
            if (num > capacity) {
                return false;
            }
        }
        return true;
    }
}

你可能感兴趣的:(笔记,#,算法分析与设计,leetcode,算法,java)