航班预订统计--差分数组解决

Posted 2021-09-15 C_YCBX Py_YYDS

文章目录

• • 题目
  • 题目解析
  • 解题代码

题目

oj平台

题目解析

这题拿到后，我们很快想到可以直接暴力累加，而这道题难度是中等，按LC这尿性，难度中等的怎么可能给你这么简单就过去了，事实也是确实如此，然后我看了看评论区，看到了差分二字，瞬间想到曾经做过一道差分的题目：检查区域是否被覆盖

• 看完之前做过的题目的文章后，瞬间就有了思路，这不就差分吗，会写后就很简单，只要构建好整个区间内的左右值，就能通过在外面循环一次得到累加结果。

解题代码

C++代码

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

Golang代码：

func corpFlightBookings(bookings [][]int, n int) []int {
    res := make([]int,n+1);
    for _,t:=range bookings{
        res[t[0]-1] += t[2]
        res[t[1]] -= t[2]
    }
    for i,_:=range res{
        if i!=0{
            res[i] += res[i-1] 
        }
    }
    return res[:len(res)-1]
}