comments | difficulty | edit_url | tags | |
---|---|---|---|---|
true |
困难 |
|
表: Buses
+--------------+------+ | Column Name | Type | +--------------+------+ | bus_id | int | | arrival_time | int | | capacity | int | +--------------+------+ bus_id 包含唯一的值。 该表的每一行都包含关于公交车到达 LeetCode 站点的时间和它的容量 (空座位的数量) 的信息。 不会出现两辆公交车同时到达,所有公交车的容量都是正整数。
表: Passengers
+--------------+------+ | Column Name | Type | +--------------+------+ | passenger_id | int | | arrival_time | int | +--------------+------+ passenger_id 包含唯一的值。 该表的每一行都包含乘客到达 LeetCode 站的时间信息。
公交车和乘客到达 LeetCode 站。如果一辆公交车在 tbus
时间点到达车站,乘客在 tpassenger
到达车站,其中 tpassenger <= tbus
,而该乘客没有赶上任何公交车,则该乘客将搭乘该公交车。此外,每辆公交车都有一个容量。如果在公交车到站的那一刻,等待的乘客超过了它的载客量 capacity
,只有 capacity
个乘客才会搭乘该公交车。
编写解决方案,报告使用每条总线的用户数量。
返回按 bus_id
升序排序 的结果表。
结果格式如下所示。
示例 1:
输入: Buses 表: +--------+--------------+----------+ | bus_id | arrival_time | capacity | +--------+--------------+----------+ | 1 | 2 | 1 | | 2 | 4 | 10 | | 3 | 7 | 2 | +--------+--------------+----------+ Passengers 表: +--------------+--------------+ | passenger_id | arrival_time | +--------------+--------------+ | 11 | 1 | | 12 | 1 | | 13 | 5 | | 14 | 6 | | 15 | 7 | +--------------+--------------+ 输出: +--------+----------------+ | bus_id | passengers_cnt | +--------+----------------+ | 1 | 1 | | 2 | 1 | | 3 | 2 | +--------+----------------+ 解释: - 11 号乘客在时间 1 到达。 - 12 号乘客在时间 1 到达。 - 1 号公交车到达时间为 2,因为有一个空座位,所以搭载了 11 号乘客。 - 2 号公交车在时间 4 到达,搭载了12 号乘客,因为它有 10 个空座位。 - 13 号乘客在时间 5 到达。 - 14 号乘客在时间 6 到达。 - 15 号乘客在时间 7 到达。 - 3 号公交车在时间 7 到达,车上有两个空座位,搭载了 12 号和 13 号乘客。
# Write your MySQL query statement below
WITH
T AS (
SELECT
*,
SUM(cnt) OVER (ORDER BY dt, bus_id) AS cur,
IF(@t > 0, @t := cnt, @t := @t + cnt) AS cur_sum
FROM
(
SELECT bus_id, arrival_time AS dt, capacity AS cnt FROM Buses
UNION ALL
SELECT -1, arrival_time AS dt, -1 FROM Passengers
) AS a JOIN (SELECT @t := 0 x) AS b
)
SELECT
bus_id,
IF(cur_sum > 0, cnt - cur_sum, cnt) AS passengers_cnt
FROM T
WHERE bus_id > 0
ORDER BY bus_id;