LeetCode 1109 航班预定统计[差分数组 前缀和] HERODING的LeetCode之路

Posted 2021-09-15 HERODING23

解题思路：
乍看这题，不就是暴力吗，怎么还是中等难度，自信满满写完代码，一跑竟然超时了，代码如下：

class Solution {
public:
    vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
        vector<int> count(n);
        for(auto& booking : bookings) {
            for(int i = booking[0] - 1; i <= booking[1] - 1; i ++) {
                count[i] += booking[2];
            }
        }
        return count;
    }
};

差分+前缀和的方法才是正道，而且并没有空间换时间，只不过是多用了一次O(n)时间而已，差分是标记每个范围的两端，最左边+当前的预定数，表示从这开始到结束都是该数，右端+1位置-当前预定数，表示结束了不用再重复加了，最后for循环遍历差分数组，让count[i] += count[i- 1];就可以实现预定统计了，代码如下：

class Solution {
public:
    vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
        vector<int> count(n);
        for(auto& booking : bookings) {
            count[booking[0] - 1] += booking[2];
            if(booking[1] < n) {
               count[booking[1]] -= booking[2];
            }
        }
        for(int i = 1; i < n; i ++) {
            count[i] += count[i- 1];
        }
        return count;
    }
};