LC 1789 [E] Primary Department for Each Employee - ALawliet/algorithms GitHub Wiki

SELECT employee_id, department_id 
FROM Employee
WHERE primary_flag = 'Y'
UNION
SELECT employee_id, department_id 
FROM Employee 
GROUP BY employee_id
HAVING COUNT(employee_id) = 1