算法题: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为例,数组变化如下:
乘客情况

使用差分数组修改上图所示的数组,结果如下:
修改为差分数组



代码(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;
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值