暴力法
class Solution {
public int[] corpFlightBookings(int[][] bookings, int n) {
int[] ans = new int[n];
for (int i = 0; i < bookings.length; i++) {
for (int j = bookings[i][0] - 1; j < bookings[i][1]; j++) {
ans[j]+= bookings[i][2];
}
}
return ans;
}
}
公交车算法
将问题转换为:某公交车共有 n 站,第 i 条记录 bookings[i] = [i, j, k] 表示在 i 站上车 k 人,乘坐到 j 站,在 j+1 站下车,需要按照车站顺序返回每一站车上的人数根据 1 的思路,定义 counter[] 数组记录每站的人数变化,counter[i] 表示第 i+1 站。遍历 bookings[]:bookings[i] = [i, j, k] 表示在 i 站增加 k 人即 counters[i-1] += k,在 j+1 站减少 k 人即 counters[j] -= k 遍历(整理)counter[] 数组,得到每站总人数: 每站的人数为前一站人数加上当前人数变化 counters[i] += counters[i - 1]
class Solution {
public int[] corpFlightBookings(int[][] bookings, int n) {
int[] counter = new int[n];
for (int[] book : bookings) {
counter[book[0]-1] += book[2];
if (book[1] < n) {
counter[book[1]] -= book[2];
}
}
for (int i = 1; i < n; i++) {
counter[i] += counter[i-1];
}
return counter;
}
}
转载请注明原文地址:https://blackberry.8miu.com/read-2790.html