1109. 航班预订统计
😀最近新创建了个开源仓库,总结LeetCode的每日一题,目前已有C++、JavaScript语言版本,欢迎大家提供其他语言版本! 🩲仓库地址:每日一题系列
题目描述:
这里有n
个航班,它们分别从1
到n
进行编号。
有一份航班预订表bookings
,表中第i
条预订记录bookings[i] = [firsti, lasti, seatsi]
意味着在从firsti
到lasti
(包含firsti
和lasti
)的每个航班上预订了seatsi
个座位。
请你返回一个长度为n
的数组answer
,其中answer[i]
是航班i
上预订的座位总数。
示例 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]
解答:
C++:
图解差分数组
class Solution {
public:
vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
vector<int> ans(n);
int len=bookings.size();
for(int i=0;i<len;++i){
ans[bookings[i][0]-1]+=bookings[i][2];
if(bookings[i][1]<n){
ans[bookings[i][1]]-=bookings[i][2];
}
}
for(int i=1;i<len;++i){
ans[i]+=ans[i-1];
}
return ans;
}
};