航班预订统计

问题描述:
这里有 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]

提示:

  • 1 <= bookings.length <= 20000
  • 1 <= bookings[i][0] <= bookings[i][1] <= n <= 20000
  • 1 <= bookings[i][2] <= 10000

解题思路:第一次看到这个题,直接就是遍历一遍,想法很好,可是LeetCode很残酷,直接超时,下面直接说优化的代码:

代码如下:

class Solution {
    public int[] corpFlightBookings(int[][] arr, int n) {
        int[] array = new int[n];
        for(int i = 0;i<arr.length;i++){
            array[arr[i][0]-1] += arr[i][2];
            if(arr[i][1]<n) array[arr[i][1]] -= arr[i][2];
        }
        for(int i = 1;i<n;i++){
            array[i] += array[i-1];
        }
        return array;
    }
}
原文地址:https://www.cnblogs.com/du001011/p/11148225.html