题目链接:https://leetcode-cn.com/contest/weekly-contest-144/problems/corporate-flight-bookings/
1.题目
这里有 n 个航班,它们分别从 1 到 n 进行编号。
我们这儿有一份航班预订表,表中第 i 条预订记录 bookings[i] = [i, j, k] 意味着我们在从 i 到 j 的每个航班上预订了 k 个座位。
请你返回一个长度为 n 的数组 answer,按航班编号顺序返回每个航班上预订的座位数。
2.思路
直接暴力计算会超时,所以用前缀和数组计算答案,只需要处理区间端点,让区间起点加预定数量,终点下一位减预定数量,最后让ans[i] = ans[i] + ans[i-1] (i>=1)
3.代码
class Solution
{
public:
vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n)
{
vector<int>ans(n);
for(int i = 0; i < bookings.size(); i++)
{
int l = bookings[i][0];
int r = bookings[i][1];
int c = bookings[i][2];
ans[l-1] += c;
if(r<n)
ans[r] -= c;
}
for(int i = 1; i < n; i++)
{
ans[i]=ans[i-1]+ans[i];
}
return ans;
}
};