Question

601. 体育馆的人流量

English Version

题目描述

表：Stadium

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

 

编写解决方案找出每行的人数大于或等于 100 且 id 连续的三行或更多行记录。

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

查询结果格式如下所示。

 

示例 1:

输入：
Stadium 表:
+------+------------+-----------+
| 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     |
+------+------------+-----------+
输出：
+------+------------+-----------+
| 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 连续的记录。

解法

方法一

# Write your MySQL query statement below
WITH
  S AS (
    SELECT
      *,
      id - (ROW_NUMBER() OVER (ORDER BY id)) AS rk
    FROM Stadium
    WHERE people >= 100
  ),
  T AS (SELECT *, COUNT(1) OVER (PARTITION BY rk) AS cnt FROM S)
SELECT id, visit_date, people
FROM T
WHERE cnt >= 3
ORDER BY 1;