Skip to content

feat: add solutions to lc problem: No.3601 #4543

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Jul 2, 2025
Merged

feat: add solutions to lc problem: No.3601 #4543

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
70 changes: 69 additions & 1 deletion solution/3600-3699/3601.Find Drivers with Improved Fuel Efficiency/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -152,14 +152,82 @@ Each row represents a trip made by a driver, including the distance traveled and

<!-- solution:start -->

### 方法一
### 方法一：分组统计 + 连接查询

我们先对 `trips` 表进行分组统计，计算每个司机在上半年和下半年的平均燃油效率。然后通过连接查询将结果与 `drivers` 表关联，筛选出燃油效率有提升的司机，并计算提升的幅度。

<!-- tabs:start -->

#### MySQL

```sql
# Write your MySQL query statement below
WITH
T AS (
SELECT
driver_id,
AVG(distance_km / fuel_consumed) half_avg,
CASE
WHEN MONTH(trip_date) <= 6 THEN 1
ELSE 2
END half
FROM trips
GROUP BY driver_id, half
)
SELECT
t1.driver_id,
d.driver_name,
ROUND(t1.half_avg, 2) first_half_avg,
ROUND(t2.half_avg, 2) second_half_avg,
ROUND(t2.half_avg - t1.half_avg, 2) efficiency_improvement
FROM
T t1
JOIN T t2 ON t1.driver_id = t2.driver_id AND t1.half < t2.half AND t1.half_avg < t2.half_avg
JOIN drivers d ON t1.driver_id = d.driver_id
ORDER BY efficiency_improvement DESC, d.driver_name;
```

#### Pandas

```python
import pandas as pd


def find_improved_efficiency_drivers(
drivers: pd.DataFrame, trips: pd.DataFrame
) -> pd.DataFrame:
trips = trips.copy()
trips["trip_date"] = pd.to_datetime(trips["trip_date"])
trips["half"] = trips["trip_date"].dt.month.apply(lambda m: 1 if m <= 6 else 2)
trips["efficiency"] = trips["distance_km"] / trips["fuel_consumed"]
half_avg = (
trips.groupby(["driver_id", "half"])["efficiency"]
.mean()
.reset_index(name="half_avg")
)
pivot = half_avg.pivot(index="driver_id", columns="half", values="half_avg").rename(
columns={1: "first_half_avg", 2: "second_half_avg"}
)
pivot = pivot.dropna()
pivot = pivot[pivot["second_half_avg"] > pivot["first_half_avg"]]
pivot["efficiency_improvement"] = (
pivot["second_half_avg"] - pivot["first_half_avg"]
).round(2)
pivot["first_half_avg"] = pivot["first_half_avg"].round(2)
pivot["second_half_avg"] = pivot["second_half_avg"].round(2)
result = pivot.reset_index().merge(drivers, on="driver_id")
result = result.sort_values(
by=["efficiency_improvement", "driver_name"], ascending=[False, True]
)
return result[
[
"driver_id",
"driver_name",
"first_half_avg",
"second_half_avg",
"efficiency_improvement",
]
]
```

<!-- tabs:end -->
Expand Down
72 changes: 71 additions & 1 deletion solution/3600-3699/3601.Find Drivers with Improved Fuel Efficiency/README_EN.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -152,14 +152,84 @@ Each row represents a trip made by a driver, including the distance traveled and

<!-- solution:start -->

### Solution 1
### Solution 1: Group Aggregation + Join Query

First, we perform group aggregation on the `trips` table to calculate the average fuel efficiency for each driver in the first half and the second half of the year.

Then, we join the results with the `drivers` table, filter out the drivers whose fuel efficiency has improved, and calculate the amount of improvement.

<!-- tabs:start -->

#### MySQL

```sql
# Write your MySQL query statement below
WITH
T AS (
SELECT
driver_id,
AVG(distance_km / fuel_consumed) half_avg,
CASE
WHEN MONTH(trip_date) <= 6 THEN 1
ELSE 2
END half
FROM trips
GROUP BY driver_id, half
)
SELECT
t1.driver_id,
d.driver_name,
ROUND(t1.half_avg, 2) first_half_avg,
ROUND(t2.half_avg, 2) second_half_avg,
ROUND(t2.half_avg - t1.half_avg, 2) efficiency_improvement
FROM
T t1
JOIN T t2 ON t1.driver_id = t2.driver_id AND t1.half < t2.half AND t1.half_avg < t2.half_avg
JOIN drivers d ON t1.driver_id = d.driver_id
ORDER BY efficiency_improvement DESC, d.driver_name;
```

#### Pandas

```python
import pandas as pd


def find_improved_efficiency_drivers(
drivers: pd.DataFrame, trips: pd.DataFrame
) -> pd.DataFrame:
trips = trips.copy()
trips["trip_date"] = pd.to_datetime(trips["trip_date"])
trips["half"] = trips["trip_date"].dt.month.apply(lambda m: 1 if m <= 6 else 2)
trips["efficiency"] = trips["distance_km"] / trips["fuel_consumed"]
half_avg = (
trips.groupby(["driver_id", "half"])["efficiency"]
.mean()
.reset_index(name="half_avg")
)
pivot = half_avg.pivot(index="driver_id", columns="half", values="half_avg").rename(
columns={1: "first_half_avg", 2: "second_half_avg"}
)
pivot = pivot.dropna()
pivot = pivot[pivot["second_half_avg"] > pivot["first_half_avg"]]
pivot["efficiency_improvement"] = (
pivot["second_half_avg"] - pivot["first_half_avg"]
).round(2)
pivot["first_half_avg"] = pivot["first_half_avg"].round(2)
pivot["second_half_avg"] = pivot["second_half_avg"].round(2)
result = pivot.reset_index().merge(drivers, on="driver_id")
result = result.sort_values(
by=["efficiency_improvement", "driver_name"], ascending=[False, True]
)
return result[
[
"driver_id",
"driver_name",
"first_half_avg",
"second_half_avg",
"efficiency_improvement",
]
]
```

<!-- tabs:end -->
Expand Down
38 changes: 38 additions & 0 deletions solution/3600-3699/3601.Find Drivers with Improved Fuel Efficiency/Solution.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
import pandas as pd


def find_improved_efficiency_drivers(
drivers: pd.DataFrame, trips: pd.DataFrame
) -> pd.DataFrame:
trips = trips.copy()
trips["trip_date"] = pd.to_datetime(trips["trip_date"])
trips["half"] = trips["trip_date"].dt.month.apply(lambda m: 1 if m <= 6 else 2)
trips["efficiency"] = trips["distance_km"] / trips["fuel_consumed"]
half_avg = (
trips.groupby(["driver_id", "half"])["efficiency"]
.mean()
.reset_index(name="half_avg")
)
pivot = half_avg.pivot(index="driver_id", columns="half", values="half_avg").rename(
columns={1: "first_half_avg", 2: "second_half_avg"}
)
pivot = pivot.dropna()
pivot = pivot[pivot["second_half_avg"] > pivot["first_half_avg"]]
pivot["efficiency_improvement"] = (
pivot["second_half_avg"] - pivot["first_half_avg"]
).round(2)
pivot["first_half_avg"] = pivot["first_half_avg"].round(2)
pivot["second_half_avg"] = pivot["second_half_avg"].round(2)
result = pivot.reset_index().merge(drivers, on="driver_id")
result = result.sort_values(
by=["efficiency_improvement", "driver_name"], ascending=[False, True]
)
return result[
[
"driver_id",
"driver_name",
"first_half_avg",
"second_half_avg",
"efficiency_improvement",
]
]
24 changes: 24 additions & 0 deletions solution/3600-3699/3601.Find Drivers with Improved Fuel Efficiency/Solution.sql
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
# Write your MySQL query statement below
WITH
T AS (
SELECT
driver_id,
AVG(distance_km / fuel_consumed) half_avg,
CASE
WHEN MONTH(trip_date) <= 6 THEN 1
ELSE 2
END half
FROM trips
GROUP BY driver_id, half
)
SELECT
t1.driver_id,
d.driver_name,
ROUND(t1.half_avg, 2) first_half_avg,
ROUND(t2.half_avg, 2) second_half_avg,
ROUND(t2.half_avg - t1.half_avg, 2) efficiency_improvement
FROM
T t1
JOIN T t2 ON t1.driver_id = t2.driver_id AND t1.half < t2.half AND t1.half_avg < t2.half_avg
JOIN drivers d ON t1.driver_id = d.driver_id
ORDER BY efficiency_improvement DESC, d.driver_name;