Sync LeetCode submission Runtime - 529 ms (47.46%), Memory - 62.6 MB (99.96%)

Author: jimit105

0184-department-highest-salary/README.md

@@ -0,0 +1,71 @@
+<p>Table: <code>Employee</code></p>
+
+<pre>
++--------------+---------+
+| Column Name  | Type    |
++--------------+---------+
+| id           | int     |
+| name         | varchar |
+| salary       | int     |
+| departmentId | int     |
++--------------+---------+
+id is the primary key (column with unique values) for this table.
+departmentId is a foreign key (reference columns) of the ID from the <code>Department </code>table.
+Each row of this table indicates the ID, name, and salary of an employee. It also contains the ID of their department.
+</pre>
+
+<p>&nbsp;</p>
+
+<p>Table: <code>Department</code></p>
+
+<pre>
++-------------+---------+
+| Column Name | Type    |
++-------------+---------+
+| id          | int     |
+| name        | varchar |
++-------------+---------+
+id is the primary key (column with unique values) for this table. It is guaranteed that department name is not <code>NULL.</code>
+Each row of this table indicates the ID of a department and its name.
+</pre>
+
+<p>&nbsp;</p>
+
+<p>Write a solution to find employees who have the highest salary in each of the departments.</p>
+
+<p>Return the result table in <strong>any order</strong>.</p>
+
+<p>The result format is in the following example.</p>
+
+<p>&nbsp;</p>
+<p><strong class="example">Example 1:</strong></p>
+
+<pre>
+<strong>Input:</strong> 
+Employee table:
++----+-------+--------+--------------+
+| id | name  | salary | departmentId |
++----+-------+--------+--------------+
+| 1  | Joe   | 70000  | 1            |
+| 2  | Jim   | 90000  | 1            |
+| 3  | Henry | 80000  | 2            |
+| 4  | Sam   | 60000  | 2            |
+| 5  | Max   | 90000  | 1            |
++----+-------+--------+--------------+
+Department table:
++----+-------+
+| id | name  |
++----+-------+
+| 1  | IT    |
+| 2  | Sales |
++----+-------+
+<strong>Output:</strong> 
++------------+----------+--------+
+| Department | Employee | Salary |
++------------+----------+--------+
+| IT         | Jim      | 90000  |
+| Sales      | Henry    | 80000  |
+| IT         | Max      | 90000  |
++------------+----------+--------+
+<strong>Explanation:</strong> Max and Jim both have the highest salary in the IT department and Henry has the highest salary in the Sales department.
+</pre>

0184-department-highest-salary/solution.py

@@ -0,0 +1,10 @@
+import pandas as pd
+
+def department_highest_salary(employee: pd.DataFrame, department: pd.DataFrame) -> pd.DataFrame:
+  df = employee.merge(department, left_on='departmentId', right_on='id', how='left')
+  df.rename(columns={'name_x': 'Employee', 'name_y': 'Department', 'salary': 'Salary'}, inplace=True)
+  max_salary = df.groupby('Department')['Salary'].transform('max')
+  df = df[df['Salary'] == max_salary]
+
+  return df[['Department', 'Employee', 'Salary']]
+