LeetCode 1109. 航班预订统计

题目描述

这里有 n 个航班，它们分别从 1 到 n 进行编号。

我们这儿有一份航班预订表，表中第 i 条预订记录 bookings[i] = [i, j, k] 意味着我们在从 i 到 j 的每个航班上预订了 k 个座位。

请你返回一个长度为 n 的数组 answer，按航班编号顺序返回每个航班上预订的座位数。

实例：

输入：bookings = [[1,2,10],[2,3,20],[2,5,25]], n = 5
输出：[10,55,45,25,25]

题目解析

更换一个题目思路，把题目转换为公交车上车问题，bookings[i] = [i, j, k] 表示从第i站上车j站下车的人数k，问题求每一站车辆上的人数。使用count[i] 来记录 第i站的上下车人数，前缀求和即为所求答案。

C++代码

class Solution {
public:
    vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
        vector<int> counter(n + 1, 0);
        vector<int> ans(n, 0);
        for(int i = 0; i < bookings.size(); i++) {
            counter[bookings[i][0] - 1] += bookings[i][2];
            counter[bookings[i][1]] -= bookings[i][2];
        }
        
        int sum = 0;
        for(int i = 0; i < n; i++) {
            sum += counter[i];
            ans[i] = sum;
        }
        return ans;
    }
};