【MySQL】WITH AS 用法以及 ROW_NUMBER 函数 和 自增ID 的巧用
力扣题
1、题目地址
601. 体育馆的人流量
2、模拟表
表：Stadium
| Column Name | Type |
|---|---|
| id | int |
| visit_date | date |
| people | int |
- visit_date 是该表中具有唯一值的列。
- 每日人流量信息被记录在这三列信息中：序号 (id)、日期 (visit_date)、 人流量 (people)
- 每天只有一行记录，日期随着 id 的增加而增加
3、要求
编写解决方案找出每行的人数大于或等于 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 连续的记录。
4、代码编写
1、我的写法
WITH one AS(
SELECT *, ROW_NUMBER() OVER(ORDER BY id) AS r, id - ROW_NUMBER() OVER(ORDER BY id) AS rk
FROM Stadium
WHERE people >= 100
)
SELECT id, visit_date, people
FROM one
WHERE rk IN (
SELECT rk
FROM one
GROUP BY rk
HAVING COUNT(rk) >= 3
)
| id | visit_date | people |
| -- | ---------- | ------ |
| 5 | 2017-01-05 | 145 |
| 6 | 2017-01-06 | 1455 |
| 7 | 2017-01-07 | 199 |
| 8 | 2017-01-09 | 188 |
2、解析
SELECT *, ROW_NUMBER() OVER(ORDER BY id) AS r, id - ROW_NUMBER() OVER(ORDER BY id) AS rk
FROM Stadium
WHERE people >= 100
| id | visit_date | people | r | rk |
| -- | ---------- | ------ | - | -- |
| 2 | 2017-01-02 | 109 | 1 | 1 |
| 3 | 2017-01-03 | 150 | 2 | 1 |
| 5 | 2017-01-05 | 145 | 3 | 2 |
| 6 | 2017-01-06 | 1455 | 4 | 2 |
| 7 | 2017-01-07 | 199 | 5 | 2 |
| 8 | 2017-01-09 | 188 | 6 | 2 |
会发现一个点，自增ID 减去 row_number 的值一样 是 连续的，我们只需要查找 同一个 rk 值出现大于等于 3 次 就满足条件
5、知识点
1、WITH AS
作用
1、在 SQL 查询中，经常会遇到需要 重复使用的子查询。
2、为了 简化查询语句 并 提高可读性，SQL 引入了 WITH AS 语法。
3、通过使用 WITH AS，我们可以创建临时表或视图，将子查询的结果保存起来，并在主查询中使用。
作用解析
1、简化复杂查询：当查询逻辑较为复杂或包含多个嵌套的子查询时，使用 WITH AS 可以将子查询逻辑分解成可读性更高的部分。这样可以降低查询的复杂度，并且更容易理解和维护。
示例1：假设有一个名为"orders"的表，存储了订单信息，包括订单号、客户ID和订单金额。我们想要查询每个客户的订单总金额，同时筛选出总金额大于1000的客户。使用 WITH AS 可以简化查询逻辑：
WITH customer_orders (customer_id, total_amount) AS (
SELECT customer_id, SUM(order_amount) AS total_amount
FROM orders
GROUP BY customer_id
)
SELECT customer_id, total_amount
FROM customer_orders
WHERE total_amount > 1000;
上述示例中，我们创建了名为"customer_orders"的临时表，存储了每个客户的订单总金额。在主查询中，我们可以直接引用"customer_orders"表，并进行筛选操作，使查询逻辑更加清晰。
2、提高查询性能：使用 WITH AS 可以避免在主查询中重复执行相同的子查询，从而提高查询性能。临时表的结果会被缓存，主查询只需要引用临时表即可，避免了重复计算子查询的开销。
示例2：假设我们需要查询员工表中工资高于平均工资的员工信息，并按工资降序排序。使用 WITH AS 可以避免重复计算平均工资：
WITH average_salary AS (
SELECT AVG(salary) AS avg_salary
FROM employees
)
SELECT employee_id, first_name, last_name, salary
FROM employees
WHERE salary > (SELECT avg_salary FROM average_salary)
ORDER BY salary DESC;
上述示例中，我们通过创建名为"average_salary"的临时表，保存了员工表中的平均工资。在主查询中，我们直接引用临时表中的平均工资，避免了重复计算的开销，提高了查询性能。
2、ROW_NUMBER 函数
专用窗口函数，按行数进行排序，具体用法可以看参考里面第二个链接和第三个链接
3、参考
SQL 中的 WITH AS 用法：简化查询，提高可读性
MySQL 窗口函数（Rows & Range）—— 滑动窗口函数用法
窗口函数 OVER(PARTITION BY) 详细用法 —— 语法 + 函数 + 开窗范围 ROWS 和 RANGE