题目
题目解析
这题拿到后,我们很快想到可以直接暴力累加,而这道题难度是中等,按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]
}