航班预订统计

题目描述 

这里有 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

解题思路 

  • 换一种思路理解题意,将问题转换为:某公交车共有 n 站,第 i 条记录 bookings[i] = [i, j, k] 表示在 i 站上车 k 人,乘坐到 j 站,在 j+1 站下车,需要按照车站顺序返回每一站车上的人数
 上车下车车上人数
第一站10010
第二站20+25010+55=55
第三站01055-10=45
第四站02045-20=25
第五站0025
  • 根据 1 的思路,定义 counter[] 数组记录每站的人数变化,即上车和下车人数的差值,counter[i] 表示第 i+1 站。遍历 bookings[]:bookings[i] = [i, j, k] 表示在 i 站增加 k 人即 counter[i-1] += k,在 j+1 站减少 k 人即 counter[j] -= k
  • 遍历(整理)counter[] 数组,得到每站总人数: 每站的人数为前一站人数加上当前人数变化 counters[i] += counters[i - 1]

 完整代码

class Solution {
    public int[] corpFlightBookings(int[][] bookings, int n) {
         // 定义n维数组,用于存储每站上车和下车人数的差值
        int[] counter = new int[n];

        for (int[] booking : bookings) {
            // 上车人数
            counter[booking[0] - 1] += booking[2];
            // 下车人数
            if (booking[1] < n) {
                counter[booking[1]] -= booking[2];
            }
        }
        // 遍历数组,每一站人数为上一站人数+本站上车和下车差值
        for (int i = 1; i < n; i++) {
            counter[i] += counter[i - 1];
        }
        return counter;
    }
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值