这里有 n 个航班,它们分别从 1 到 n 进行编号。
我们这儿有一份航班预订表,表中第 i 条预订记录 bookings[i] = [j, k, l] 意味着我们在从 j 到 k 的每个航班上预订了 l 个座位。
请你返回一个长度为 n 的数组 answer,按航班编号顺序返回每个航班上预订的座位数。
示例:
输入:bookings = [[1,2,10],[2,3,20],[2,5,25]], n = 5
输出:[10,55,45,25,25]
提示:
1 <= bookings.length <= 20000
1 <= bookings[i][0] <= bookings[i][1] <= n <= 20000
1 <= bookings[i][2] <= 10000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/corporate-flight-bookings
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:暴力法是很容易想到的,但是超时。我们需要的是使用前缀和的方法,利用公交车上车下车的算法,ans
中首先存储的是每一个站的变化量,而初始站是为0
的,所以我们从初始状态开始累加变化量就是每个站最终的人数。
class Solution {
public:
vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
vector<int>ans(n,0);
int rows = bookings.size();
if(rows == 0)
return ans;
int colmns = bookings[0].size();
for(int i = 0;i < rows;++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 < n;++i)
{
ans[i] +=ans[i-1];//变化量累计求和
}
return ans;
}
};