1789. 员工的直属部门

1789. 员工的直属部门

表：Employee

±--------------±--------+
| Column Name | Type |
±--------------±--------+
| employee_id | int |
| department_id | int |
| primary_flag | varchar |
±--------------±--------+

这张表的主键为 employee_id, department_id (具有唯一值的列的组合) employee_id 是员工的ID
department_id 是部门的ID，表示员工与该部门有关系 primary_flag 是一个枚举类型，值分别为(‘Y’, ‘N’).
如果值为’Y’,表示该部门是员工的直属部门。 如果值是’N’,则否
一个员工可以属于多个部门。当一个员工加入超过一个部门的时候，他需要决定哪个部门是他的直属部门。请注意，当员工只加入一个部门的时候，那这个部门将默认为他的直属部门，虽然表记录的值为’N’.

请编写解决方案，查出员工所属的直属部门。
返回结果 没有顺序要求 。

返回结果格式如下例子所示：
示例 1：

输入：
Employee table:
±------------±--------------±-------------+
| employee_id | department_id | primary_flag |
±------------±--------------±-------------+
| 1 | 1 | N |
| 2 | 1 | Y |
| 2 | 2 | N |
| 3 | 3 | N |
| 4 | 2 | N |
| 4 | 3 | Y |
| 4 | 4 | N |
±------------±--------------±-------------+
输出：
±------------±--------------+
| employee_id | department_id |
±------------±--------------+
| 1 | 1 |
| 2 | 1 |
| 3 | 3 |
| 4 | 3 |
±------------±--------------+
解释：

• 员工 1 的直属部门是 1
• 员工 2 的直属部门是 1
• 员工 3 的直属部门是 3
• 员工 4 的直属部门是 3

解题方法

1、对于每个员工，查询会返回唯一的 department_id。如果员工只属于一个部门，直接返回该部门；如果属于多个部门，则返回 primary_flag = ‘Y’ 对应的主要部门。

select
employee_id,
if(
    count(department_id)=1,
    department_id,
    max(case primary_flag when 'Y' then department_id end
)) as department_id
from Employee
group by employee_id

2、对于每个员工，如果有 primary_flag = ‘Y’，返回该部门。如果员工只属于一个部门（无论 primary_flag 是 Y 还是 N），也会返回该部门。

select employee_id, department_id
from employee 
where primary_flag="Y"
or
employee_id in (
    select employee_id
    from Employee
    group by employee_id
    having count(employee_id)=1
)

如果数据较为复杂且需要处理多个部门，推荐使用 查询 1。