Project Employees II — LeetCode 1076 Python Solution

Table: Project +-------------+---------+ | Column Name | Type | +-------------+---------+ | project_id | int | | employee_id | int | +-------------+---------+ (project_id, employee_id) is the primary key (combination of columns with unique values) of this table. employee_id is a foreign key (reference column) to Employee table.

+-------------+---------+
| Column Name | Type    |
+-------------+---------+
| project_id  | int     |
| employee_id | int     |
+-------------+---------+
(project_id, employee_id) is the primary key (combination of columns with unique values) of this table.
employee_id is a foreign key (reference column) to Employee table.
Each row of this table indicates that the employee with employee_id is working on the project with project_id.

import duckdb
import pandas as pd

def solution(project: pd.DataFrame, employee: pd.DataFrame) -> pd.DataFrame:
    con = duckdb.connect()
    con.register("Project", project)
    con.register("Employee", employee)
    return con.execute("""SELECT project_id
FROM Project
GROUP BY 1
HAVING
    COUNT(1) >= all(
        SELECT COUNT(1)
        FROM Project
        GROUP BY project_id
    );""").df()