【每日一题】1109. 航班预订统计

1109. 航班预订统计 - 力扣（LeetCode）

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

有一份航班预订表 bookings ，表中第 i 条预订记录 bookings[i] = [firsti, lasti, seatsi] 意味着在从 firsti 到 lasti （包含 firsti 和 lasti ）的 每个航班 上预订了 seatsi 个座位。

请你返回一个长度为 n 的数组 answer，里面的元素是每个航班预定的座位总数。

示例 1：

输入：bookings = [[1,2,10],[2,3,20],[2,5,25]], n = 5
输出：[10,55,45,25,25]
解释：
航班编号        1   2   3   4   5
预订记录 1 ：   10  10
预订记录 2 ：       20  20
预订记录 3 ：       25  25  25  25
总座位数：      10  55  45  25  25
因此，answer = [10,55,45,25,25]

示例 2：

输入：bookings = [[1,2,10],[2,2,15]], n = 2
输出：[10,25]
解释：
航班编号        1   2
预订记录 1 ：   10  10
预订记录 2 ：       15
总座位数：      10  25
因此，answer = [10,25]

 

class Solution {
    public int[] corpFlightBookings(int[][] bookings, int n) {
        int[] dif = new int[n+1];
        int[] nums = new int[n];

        for(int i = 0; i < n ; i++) {
            dif[i] = 0; 
            nums[i] = 0;
        }

        int len = bookings.length;
        
        for(int i = 0 ; i < len ; i++) {
            int num = bookings[i][2];
            int index1 = bookings[i][0]-1;
            int index2 = bookings[i][1];

            dif[index1] += num;
            dif[index2] -= num;
        }

        nums[0] = dif[0] ;

        for(int i = 1 ; i < n ; i++){
            nums[i] = dif[i] + nums[i-1];
        }

        return nums;
    }
}

        每日一题，今天是中等题。这道题是标准的差分数组题，之后也会整理在零基础算法里面。

        读题。实际上就是有n个航班，每个航班会有人定位置，有人不止定一个航班的位置。实际上就是对一个初始全为0的数组进行加减运算而已。如果暴力解题的话，就是每次根据给所给的范围，给数组的每个位置进行加减，但我们可以使用差分数组来解题。

        差分数组实际上和前缀和差不多的思路，只是每个位置变成了这个位置和前一个位置的差。即dif[i] = num[i] - num[i-1]。所以，求解num[i]时可以得到：num[i]=dif[i]+num[i-1]。那么，如果你给dif数组的 i 位置加上一个数会怎么样？由上面的公式实际上可以知道，原数组i位置及其后面的位置都会加上这个数，相当于num[i]=dif[i]+num[i-1]+C（常数）。那减法就相通，相当于这个位置及其后面的位置都减去这个数字即：num[i]=dif[i]+num[i-1]-C（常数）。那如果我们给i位置加上一个数，给j（j>i）位置减去一个数，就相当于原数组[i，j)之间的所有数都加上了这个数。

        依据上述的所说，我们只需要构造一个dif的差分数组，就可以轻松地进行修改，最后只需要依据原数组（该题中全为0）以及差分数组，一次遍历就可以得到答案了。

        最后情况如下：