最后一个能进入电梯的人

1204. 最后一个能进入电梯的人

表: Queue

+-------------+---------+
| Column Name | Type    |
+-------------+---------+
| person_id   | int     |
| person_name | varchar |
| weight      | int     |
| turn        | int     |
+-------------+---------+
person_id 是这个表的主键。
该表展示了所有等待电梯的人的信息。
表中 person_id 和 turn 列将包含从 1 到 n 的所有数字，其中 n 是表中的行数。

有一群人在等着上公共汽车。然而，巴士有1000 公斤的重量限制，所以可能会有一些人不能上。

写一条 SQL 查询语句查找 最后一个 能进入电梯且不超过重量限制的 person_name 。题目确保队列中第一位的人可以进入电梯，不会超重。

查询结果如下所示。

示例 1:

输入：
Queue 表
+-----------+-------------------+--------+------+
| person_id | person_name       | weight | turn |
+-----------+-------------------+--------+------+
| 5         | George Washington | 250    | 1    |
| 3         | John Adams        | 350    | 2    |
| 6         | Thomas Jefferson  | 400    | 3    |
| 2         | Will Johnliams    | 200    | 4    |
| 4         | Thomas Jefferson  | 175    | 5    |
| 1         | James Elephant    | 500    | 6    |
+-----------+-------------------+--------+------+
输出：
+-------------------+
| person_name       |
+-------------------+
| Thomas Jefferson  |
+-------------------+
解释：
为了简化，Queue 表按 turn 列由小到大排序。
上例中 George Washington(id 5), John Adams(id 3) 和 Thomas Jefferson(id 6) 将可以进入电梯,因为他们的体重和为 250 + 350 + 400 = 1000。
Thomas Jefferson(id 6) 是最后一个体重合适并进入电梯的人。

原站题解

去查看

上次编辑到这里，代码来自缓存点击恢复默认模板

# Write your MySQL query statement below

pythondata 解法, 执行用时: 384 ms, 内存消耗: 65.7 MB, 提交时间: 2024-05-27 11:40:03

import pandas as pd

def last_passenger(queue: pd.DataFrame) -> pd.DataFrame:
    # 通过turn排序
    queue = queue.sort_values(by = 'turn')
    # 通过cumsum计算累加求和
    queue['sum_weight'] = queue['weight'].cumsum()
    # 筛选出最后一个上车的乘客
    return queue.loc[queue.sum_weight <= 1000].tail(1)[['person_name']]

mysql 解法, 执行用时: 1608 ms, 内存消耗: 0 B, 提交时间: 2023-04-02 12:07:35

# Write your MySQL query statement below
select person_name
from Queue q1
where (select sum(weight) from Queue where turn <= q1.turn) <= 1000
order by turn desc limit 1;

mysql 解法, 执行用时: 558 ms, 内存消耗: 0 B, 提交时间: 2023-04-02 12:07:20

# Write your MySQL query statement below
select 
    person_name
from
(select 
    turn,
    person_name,
    sum(weight) over(order by turn) as cumu_weight
from Queue) t
where cumu_weight <= 1000
order by turn desc
limit 1;

mysql 解法, 执行用时: 574 ms, 内存消耗: 0 B, 提交时间: 2023-04-02 12:07:03

# Write your MySQL query statement below
SELECT a.person_name
FROM (
	SELECT person_name, @pre := @pre + weight AS weight
	FROM Queue, (SELECT @pre := 0) tmp
	ORDER BY turn
) a
WHERE a.weight <= 1000
ORDER BY a.weight DESC
LIMIT 1;

mysql 解法, 执行用时: 1211 ms, 内存消耗: 0 B, 提交时间: 2023-04-02 12:06:50

# Write your MySQL query statement below
SELECT a.person_name
FROM Queue a, Queue b
WHERE a.turn >= b.turn
GROUP BY a.person_id HAVING SUM(b.weight) <= 1000
ORDER BY a.turn DESC
LIMIT 1;

上一题

下一题

详情