Question

177. 第 N 高的薪水

English Version

题目描述

表: Employee

+-------------+------+
| Column Name | Type |
+-------------+------+
| id      | int  |
| salary    | int  |
+-------------+------+
在 SQL 中，id 是该表的主键。
该表的每一行都包含有关员工工资的信息。

 

查询 Employee 表中第 n 高的工资。如果没有第 n 个最高工资，查询结果应该为 null 。

查询结果格式如下所示。

 

示例 1:

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

示例 2:

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

解法

方法一：排序 + LIMIT

我们可以先对 salary 进行降序排序，然后使用 LIMIT 语句获取第 $n$ 高的工资。

import pandas as pd


def nth_highest_salary(employee: pd.DataFrame, N: int) -> pd.DataFrame:
  unique_salaries = employee.salary.unique()
  if len(unique_salaries) < N:
    return pd.DataFrame([np.NaN], columns=[f"getNthHighestSalary({N})"])
  else:
    salary = sorted(unique_salaries, reverse=True)[N - 1]
    return pd.DataFrame([salary], columns=[f"getNthHighestSalary({N})"])

CREATE FUNCTION getNthHighestSalary(N INT) RETURNS INT
BEGIN
  SET N = N - 1;
  RETURN (
    # Write your MySQL query statement below.
    SELECT (
      SELECT DISTINCT salary
      FROM Employee
      ORDER BY salary DESC
      LIMIT 1 OFFSET N
    )
  );
END