[leetCode]1109. 航班预订统计

暴力法

class Solution {
    public int[] corpFlightBookings(int[][] bookings, int n) {
        int[] ans = new int[n];
        for (int i = 0; i < bookings.length; i++) {
            for (int j = bookings[i][0] - 1; j < bookings[i][1]; j++) {
                ans[j]+= bookings[i][2];
            }
        }
        return ans;
    }
}

公交车算法

1. 将问题转换为：某公交车共有 n 站，第 i 条记录 bookings[i] = [i, j, k] 表示在 i 站上车 k 人，乘坐到 j 站，在 j+1 站下车，需要按照车站顺序返回每一站车上的人数
2. 根据 1 的思路，定义 counter[] 数组记录每站的人数变化，counter[i] 表示第 i+1 站。遍历 bookings[]：bookings[i] = [i, j, k] 表示在 i 站增加 k 人即 counters[i-1] += k，在 j+1 站减少 k 人即 counters[j] -= k
   遍历(整理)counter[] 数组，得到每站总人数： 每站的人数为前一站人数加上当前人数变化 counters[i] += counters[i - 1]

class Solution {
    public int[] corpFlightBookings(int[][] bookings, int n) {
        int[] counter = new int[n];
        for (int[] book : bookings) {
            counter[book[0]-1] += book[2];
            if (book[1] < n) {
                counter[book[1]] -= book[2];
            }
        }
        for (int i = 1; i < n; i++) {
            counter[i] += counter[i-1];
        }
        return counter;
    }
}