LeetCode每日一题 | LeetCode-1094.拼车

LeetCode-1094.拼车

    • 题目描述
    • 问题分析
    • 程序代码

题目描述

原题链接

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

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

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

问题分析

由于车的位置范围为[0, 1000]。因此,我们可以使用一个长度为 1000 的数组来记录每个位置的乘客数量。

先遍历trips数组,利用差分数组的思想,修改某段区间。即若有 c 个乘客在 a 点上车,在 b 点下车,要对区间[a, b)整体进行加 c 的操作,利用差分数组只需要进行nums[a] += cnums[b] -= c操作即可。

求完差分数组后,对差分数组进行前缀和计算,就可以得到每个站点的乘客数量,与车的最大容量进行比较便可得到最终答案。

程序代码

class Solution {
public:
    bool carPooling(vector<vector<int>>& trips, int capacity) {
        vector<int> nums(1010, 0);
        for(auto t : trips) {
            nums[t[1]+1] += t[0];
            nums[t[2]+1] -= t[0];
        }

        for(int i = 1; i <= 1000; i++) {
            nums[i] += nums[i-1];
            if(nums[i] > capacity)  return false;
        }

        return true;
    }
};

你可能感兴趣的:(leetcode题解,leetcode,算法,职场和发展)