题目描述:
这里有 n 个航班,它们分别从 1 到 n 进行编号。
我们这儿有一份航班预订表,表中第 i 条预订记录 bookings[i] = [i, j, k] 意味着我们在从 i 到 j 的每个航班上预订了 k 个座位。
请你返回一个长度为 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
跟之前的那道上车人数差不多,你先需要将开始的站累加,然后在结束站减去,之后累加和即可,不要考虑之间的每个航班之间的具体数量,那样会超时。
代码:
class Solution {
public int[] corpFlightBookings(int[][] bookings, int n) {
int result[] = new int[n];
for (int j = 0; j < bookings.length ; j++) {
int start = bookings[j][0];
int end = bookings[j][1];
int num = bookings[j][2];
result[start - 1] += num;
if(end < n){
result[end] -= num;
}
}
for (int i = 1; i < result.length; i++) {
result[i] += result[i - 1];
}
return result;
}
}