leetcode 1811 寻找面试候选人(postgresql)

需求

表: Contests

±-------------±-----+
| Column Name | Type |
±-------------±-----+
| contest_id | int |
| gold_medal | int |
| silver_medal | int |
| bronze_medal | int |
±-------------±-----+
contest_id 是该表的主键.
该表包含LeetCode竞赛的ID和该场比赛中金牌、银牌、铜牌的用户id。
可以保证，所有连续的比赛都有连续的ID，没有ID被跳过。

Table: Users

±------------±--------+
| Column Name | Type |
±------------±--------+
| user_id | int |
| mail | varchar |
| name | varchar |
±------------±--------+
user_id 是该表的主键.
该表包含用户信息。

编写 SQL 语句来返回 所有面试候选人 的姓名 name 和邮件 mail 。当用户满足以下两个要求中的 任意一条 ，其成为 面试候选人 :

该用户在 连续三场及更多 比赛中赢得 任意 奖牌。
该用户在 三场及更多不同的 比赛中赢得 金牌（这些比赛可以不是连续的）
可以以 任何顺序 返回结果。

查询结果格式如下例所示。

示例 1：

输入：
Contests表:
±-----------±-----------±-------------±-------------+
| contest_id | gold_medal | silver_medal | bronze_medal |
±-----------±-----------±-------------±-------------+
| 190 | 1 | 5 | 2 |
| 191 | 2 | 3 | 5 |
| 192 | 5 | 2 | 3 |
| 193 | 1 | 3 | 5 |
| 194 | 4 | 5 | 2 |
| 195 | 4 | 2 | 1 |
| 196 | 1 | 5 | 2 |
±-----------±-----------±-------------±-------------+
Users表:
±--------±-------------------±------+
| user_id | mail | name |
±--------±-------------------±------+
| 1 | sarah@leetcode.com | Sarah |
| 2 | bob@leetcode.com | Bob |
| 3 | alice@leetcode.com | Alice |
| 4 | hercy@leetcode.com | Hercy |
| 5 | quarz@leetcode.com | Quarz |
±--------±-------------------±------+
输出：
±------±-------------------+
| name | mail |
±------±-------------------+
| Sarah | sarah@leetcode.com |
| Bob | bob@leetcode.com |
| Alice | alice@leetcode.com |
| Quarz | quarz@leetcode.com |
±------±-------------------+
解释：
Sarah 赢得了3块金牌 (190, 193, and 196),所以我们将她列入结果表。
Bob在连续3场竞赛中赢得了奖牌(190, 191, and 192), 所以我们将他列入结果表。
- 注意他在另外的连续3场竞赛中也赢得了奖牌(194, 195, and 196).
Alice在连续3场竞赛中赢得了奖牌 (191, 192, and 193), 所以我们将她列入结果表。
Quarz在连续5场竞赛中赢得了奖牌(190, 191, 192, 193, and 194), 所以我们将他列入结果表。

进阶：

如果第一个条件变成“该用户在连续 n 场及比赛中赢得任意奖牌。”呢？你如何更改你的解法，来选出面试候选人？可以把 n 想象成存储过程中的参数。
有的用户可能没有参加每一场竞赛，但是在参加的每一场竞赛中都表现得不错。你如何更改你的解法，以达到只考虑那些 用户参与了的 比赛？可假设另一张表给出了每场比赛的注册用户信息。

输入

输出

with t1 as (
-- 获得三块及以上的金牌的用户
select gold_medal as medal
from contests
group by gold_medal
having count(1)>=3
),t2 as (
select contest_id,gold_medal as medal
from contests
union all
select contest_id,silver_medal as medal
from contests
union all
select contest_id,bronze_medal as medal
from contests
order by medal,contest_id
),t3 as (
select *,
       (contest_id-(row_number() over (partition by medal order by contest_id))) as rn1
from t2
),t4 as (
select distinct medal
from t3
group by rn1,medal
having count(1)>=3
union all
select * from t1
)
select name,mail
from users,t4
where user_id=medal;