leetcode1109 航班预订统计
Posted JYeontu
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了leetcode1109 航班预订统计相关的知识,希望对你有一定的参考价值。
题目
这里有 n 个航班,它们分别从 1 到 n 进行编号。
有一份航班预订表 bookings ,表中第 i 条预订记录 bookings[i] = [firsti, lasti, seatsi] 意味着在从 firsti 到 lasti (包含 firsti 和 lasti )的 每个航班 上预订了 seatsi 个座位。
请你返回一个长度为 n 的数组 answer,其中 answer[i] 是航班 i 上预订的座位总数。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/probl...
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例 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]
提示:
1 <= n <= 2 * 104
1 <= bookings.length <= 2 * 104
bookings[i].length == 3
1 <= firsti <= lasti <= n
1 <= seatsi <= 104
题解
1、暴力(双层循环解题)
/**
* @param {number[][]} bookings
* @param {number} n
* @return {number[]}
*/
var corpFlightBookings = function(bookings, n) {
let res = new Array(n).fill(0);
for(let i = 0; i < bookings.length; i++){
for(let j = bookings[i][0]; j <= bookings[i][1]; j++){
res[j-1] += bookings[i][2];
}
}
return res;
};
2、差分
将区间[l,r]整体增加一个值 val操作,我们可以对差分数组arr的影响看成两部分:
(1)区间左端点:arr[l] += val,由于差分是前缀和的逆向过程,这个操作对于将来的查询而言,带来的影响是对于所有的下标大于等于 l 的位置都增加了值 val;
(2)区间右端点:arr[r + 1] -= val,由于我们期望只对区间[l,r]产生影响,因此需要对下标大于r的位置进行减值操作,右端的值就不会受到前面操作的影响。
具体代码如下
/**
* @param {number[][]} bookings
* @param {number} n
* @return {number[]}
*/
var corpFlightBookings = function(bookings, n) {
let res = new Array(n).fill(0);
for(let i = 0; i < bookings.length; i++){
//区间左端点
res[bookings[i][0] - 1] += bookings[i][2];
//区间右端点+1
if(bookings[i][1] < n){
res[bookings[i][1]] -= bookings[i][2];
}
}
for(let i = 1; i < n; i++){
res[i] += res[i - 1];
}
return res;
};
以上是关于leetcode1109 航班预订统计的主要内容,如果未能解决你的问题,请参考以下文章
Leetcode刷题100天—1109. 航班预订统计(数组)—day24