LeetCode第177题_第N高的薪水-CSDN博客

本文链接：https://blog.csdn.net/qq_40263592/article/details/147492602

LeetCode 第177题：第N高的薪水

题目描述

编写一个 SQL 查询，获取 Employee 表中第 n 高的薪水（Salary）。

+-------------+------+
| Column Name | Type |
+-------------+------+
| id          | int  |
| salary      | int  |
+-------------+------+
id 是这个表的主键。
表的每一行包含员工的工资信息。

例如，给定上表，如果 n = 2，则应该返回第二高的薪水。如果没有第 n 高的薪水，那么查询应返回 null。

难度

中等

题目链接

点击在LeetCode中查看题目

示例

示例1：

输入: 
Employee 表:
+----+--------+
| id | salary |
+----+--------+
| 1  | 100    |
| 2  | 200    |
| 3  | 300    |
+----+--------+
n = 2
输出: 
+------------------------+
| getNthHighestSalary(2) |
+------------------------+
| 200                    |
+------------------------+

示例2：

输入: 
Employee 表:
+----+--------+
| id | salary |
+----+--------+
| 1  | 100    |
+----+--------+
n = 2
输出: 
+------------------------+
| getNthHighestSalary(2) |
+------------------------+
| null                   |
+------------------------+

提示

如果没有第 n 高的薪水，查询应返回 null

解题思路

方法一：使用 DENSE_RANK() 窗口函数

MySQL 中可以使用窗口函数 DENSE_RANK() 来为每一行分配一个排名，然后筛选出排名为 n 的行。

关键点：

使用 DENSE_RANK() 函数对薪水进行降序排名
筛选出排名恰好等于 n 的行
使用子查询确保在没有第 n 高薪水时返回 NULL

时间复杂度：O(n log n)，其中 n 是 Employee 表中的行数（排序的时间复杂度）
空间复杂度：O(n)

方法二：使用 ORDER BY 和 LIMIT 子句

类似于第二高薪水的解法，我们可以使用 ORDER BY 对薪水进行降序排序，然后使用 LIMIT 和 OFFSET 来获取第 n 高的薪水。

关键点：

使用 DISTINCT 去除重复值
对薪水进行降序排序
使用 LIMIT 1 OFFSET (n-1) 获取第 n 条记录
在存储过程中需要特别注意 n-1 的计算

时间复杂度：O(n log n)，其中 n 是 Employee 表中的行数（排序的时间复杂度）
空间复杂度：O(n)

代码实现

SQL 实现（方法一：DENSE_RANK() 窗口函数）

CREATE FUNCTION getNthHighestSalary(N INT) RETURNS INT
BEGIN
  RETURN (
      SELECT 
          DISTINCT salary
      FROM 
          (SELECT salary, DENSE_RANK() OVER (ORDER BY salary DESC) AS rnk
          FROM Employee) AS ranked
      WHERE rnk = N
  );
END

SQL 实现（方法二：ORDER BY 和 LIMIT）

CREATE FUNCTION getNthHighestSalary(N INT) RETURNS INT
BEGIN
  DECLARE M INT;
  SET M = N - 1;
  RETURN (
      SELECT 
          DISTINCT salary
      FROM 
          Employee
      ORDER BY 
          salary DESC
      LIMIT 1 OFFSET M
  );
END