1109.航班预订统计
暴力解法
笔者的最初想到的思路是对每次的预订记录进行累加,遍历每个二维数组的子元素求得答案。
class Solution {
public int[] corpFlightBookings(int[][] bookings, int n) {
int[] answer = new int[n];
for(int i = 0;i < bookings.length;i++){
int t = bookings[i][0]-1;
while(t < bookings[i][1]){
answer[t] = answer[t] + bookings[i][2];
t++;
}
}
return answer;
}
}
暴力解法虽然思路清晰,但用时明显过长。(虽然通过)
差分数组
对于此题我们可以引入一个差分数组的概念。
我们拥有数组bookings,bookings[i]代表每份订单,每份订单由三部分组成:
- bookings[i][0]: 订单起始号。
- bookings[i][1]: 订单末尾号。
- bookings[i][2]: 订单人数。
我们假设新建一个长度为n的answer整型数组。answer的序号便为航班号。那么我们是不是可以对answer进行赋值,对应的航班开始处,给answer数组赋值。
answer[航班号-1] = answer[航班号-1] + bookings[i][1]
那么我们对其进行累和是不是能够得到相应位置的预定数呢?
answer[i] = answer[i] + answer[i-1]
这里我们除了在开始时给数组赋值,同样也要在结束时减去相应的值。
answer[结束时的航班号] = answer[结束时的航班号] - bookings[i][1]
//注意此处是结束时的航班号,因为对于answer数组来说,在结束时一份订单时,预订仍然存在。
完整代码:
class Solution {
public int[] corpFlightBookings(int[][] bookings, int n) {
int[] answer = new int[n];
for(int i = 0;i < bookings.length;i++){
answer[bookings[i][0] - 1] = answer[bookings[i][0] - 1] + bookings[i][2];
if(bookings[i][1] < n)
answer[bookings[i][1]] = answer[bookings[i][1]] - bookings[i][2];
}
for(int i = 1;i < n;i++){
answer[i] = answer[i] + answer[i-1];
}
return answer;
}
}
提升明显。