首页 > 数据库> > 【LeetCode-SQL】601. 体育馆的人流量

【LeetCode-SQL】601. 体育馆的人流量

2022-01-17 09:03:31 作者：互联网

一、题目

表：Stadium

+---------------+---------+
| Column Name   | Type    |
+---------------+---------+
| id            | int     |
| visit_date    | date    |
| people        | int     |
+---------------+---------+
visit_date 是表的主键
每日人流量信息被记录在这三列信息中：序号 (id)、日期 (visit_date)、 人流量 (people)
每天只有一行记录，日期随着 id 的增加而增加

编写一个 SQL 查询以找出每行的人数大于或等于 100 且 id 连续的三行或更多行记录。

返回按 visit_date 升序排列的结果表。

查询结果格式如下所示。

Stadium table:
+------+------------+-----------+
| id   | visit_date | people    |
+------+------------+-----------+
| 1    | 2017-01-01 | 10        |
| 2    | 2017-01-02 | 109       |
| 3    | 2017-01-03 | 150       |
| 4    | 2017-01-04 | 99        |
| 5    | 2017-01-05 | 145       |
| 6    | 2017-01-06 | 1455      |
| 7    | 2017-01-07 | 199       |
| 8    | 2017-01-09 | 188       |
+------+------------+-----------+

Result table:
+------+------------+-----------+
| id   | visit_date | people    |
+------+------------+-----------+
| 5    | 2017-01-05 | 145       |
| 6    | 2017-01-06 | 1455      |
| 7    | 2017-01-07 | 199       |
| 8    | 2017-01-09 | 188       |
+------+------------+-----------+
id 为 5、6、7、8 的四行 id 连续，并且每行都有 >= 100 的人数记录。
请注意，即使第 7 行和第 8 行的 visit_date 不是连续的，输出也应当包含第 8 行，因为我们只需要考虑 id 连续的记录。
不输出 id 为 2 和 3 的行，因为至少需要三条 id 连续的记录。

二、解决

1、关联查询

思路：

S1 是高峰期第 1 天：(s2.id-s1.id=1 and s3.id-s2.id=1) – s1, s2, s3
S2 是高峰期第 2 天：(s1.id-s2.id=1 and s3.id-s1.id=1) – s2, s1, s3
S3 是高峰期第 3 天：(s3.id-s2.id=1 and s1.id-s3.id=1) – s2, s3, s1

代码：

select distinct s1.* 
from stadium s1, stadium s2, stadium s3
where (s1.people>=100 and s2.people>=100 and s3.people>=100)
and (
      (s2.id-s1.id=1 and s3.id-s2.id=1)  -- s1, s2, s3
   or (s1.id-s2.id=1 and s3.id-s1.id=1)  -- s2, s1, s3
   or (s3.id-s2.id=1 and s1.id-s3.id=1)  -- s2, s3, s1
)
# group by s1.id
order by s1.id;

2、窗口函数

思路：
代码-版本1：

select id, visit_date, people
from(
    select *,count(*)over(partition by t_rank) t2_rank
    from (
        select *,id-row_number() over(order by id) t_rank
        from stadium  
        where people > 99
        ) t
    ) t2
where t2.t2_rank>2
order by id;

代码-版本2：

with t1 as(
    select *,id - row_number() over(order by id) as rk
    from stadium
    where people >= 100
)

select id,visit_date,people
from t1
where rk in(
    select rk
    from t1
    group by rk
    having count(rk) >= 3
)

三、参考

1、体育馆的人流量
2、窗口函数解法
3、图解——连续日期及难点分析