1094. 拼车

 

labuladong 题解思路

难度中等199收藏分享切换为英文接收动态反馈

车上最初有 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 <= 105

 

 

 

 

class Solution {
public:
    bool carPooling(vector<vector<int>>& trips, int capacity) {
        vector<int> res(1001,0);
        vector<int> df(1001,0);

        for(auto trip: trips) {
            int num_passenger = trip[0], from = trip[1], to = trip[2]-1;
            df[from]+=num_passenger;
            if(to+1<1001) df[to+1]-=num_passenger;
        }
        res[0] = df[0];
        for (int i = 1; i <=1000;i++) {
            res[i] = df[i] + res[i-1];
        }
        //
        for(auto x: res) {
            if (x>capacity) return false;
        }
        return true;
    }
};