Leetcode 569. 员工薪水中位数

1.题目基本信息

1.1.题目描述

表: Employee

+--------------+---------+

| Column Name | Type |

+--------------+---------+

| id | int |

| company | varchar |

| salary | int |

+--------------+---------+

id 是该表的主键列(具有唯一值的列)。

该表的每一行表示公司和一名员工的工资。

编写解决方案，找出每个公司的工资中位数。

以 任意顺序 返回结果表。

1.2.题目地址

https://leetcode.cn/problems/median-employee-salary/description/

2.解题方法

2.1.解题思路

先在with中统计每个公司的员工数量，然后使用窗口函数计算每个公司按salary升序排列的行号，然后判断每一个记录是不是中位数记录，最后筛选即可

2.2.解题步骤

请参考思路部分和注释。

3.解题代码

sql代码

# Write your MySQL query statement below
WITH cnt1 AS (
    SELECT company, COUNT(*) AS cnt FROM Employee GROUP BY company
)
# 思路：先在with中统计每个公司的员工数量，然后使用窗口函数计算每个公司按salary升序排列的行号，然后判断每一个记录是不是中位数记录，最后筛选即可
SELECT 
    id, company, salary
FROM (
    SELECT 
        id, company, salary, 
        IF(cnt % 2 = 1, IF(
            row_num = CEIL(cnt / 2), 1, 0
        ), IF(row_num = FLOOR(cnt / 2) OR row_num = FLOOR(cnt / 2) + 1, 1, 0)) AS is_mid
    FROM (
        SELECT 
            id, Employee.company, salary,
            ROW_NUMBER() OVER (PARTITION BY Employee.company ORDER BY salary) AS row_num, 
            cnt
        FROM 
            Employee LEFT JOIN cnt1 ON Employee.company = cnt1.company
    ) AS t1
) AS t2
WHERE is_mid = 1

4.执行结果