A train line going through a city has two routes, the regular route and the express route. Both routes go through the same n + 1 stops labeled from 0 to n. Initially, you start on the regular route at stop 0.
You are given two 1-indexed integer arrays regular and express, both of length n. regular[i] describes the cost it takes to go from stop i - 1 to stop i using the regular route, and express[i] describes the cost it takes to go from stop i - 1 to stop i using the express route.
You are also given an integer expressCost which represents the cost to transfer from the regular route to the express route.
Note that:
expressCost every time you transfer from the regular route to the express route.Return a 1-indexed array costs of length n, where costs[i] is the minimum cost to reach stop i from stop 0.
Note that a stop can be counted as reached from either route.
Example 1:
Input: regular = [1,6,9,5], express = [5,2,3,10], expressCost = 8 Output: [1,7,14,19] Explanation: The diagram above shows how to reach stop 4 from stop 0 with minimum cost. - Take the regular route from stop 0 to stop 1, costing 1. - Take the express route from stop 1 to stop 2, costing 8 + 2 = 10. - Take the express route from stop 2 to stop 3, costing 3. - Take the regular route from stop 3 to stop 4, costing 5. The total cost is 1 + 10 + 3 + 5 = 19. Note that a different route could be taken to reach the other stops with minimum cost.
Example 2:
Input: regular = [11,5,13], express = [7,10,6], expressCost = 3 Output: [10,15,24] Explanation: The diagram above shows how to reach stop 3 from stop 0 with minimum cost. - Take the express route from stop 0 to stop 1, costing 3 + 7 = 10. - Take the regular route from stop 1 to stop 2, costing 5. - Take the express route from stop 2 to stop 3, costing 3 + 6 = 9. The total cost is 10 + 5 + 9 = 24. Note that the expressCost is paid again to transfer back to the express route.
Constraints:
n == regular.length == express.length1 <= n <= 1051 <= regular[i], express[i], expressCost <= 105Problem Overview: You travel through n train stations. At each station you can continue on the regular line or switch to an express line that requires a one-time expressCost. Each segment has different travel costs for the regular and express lines. The task is to compute the minimum cost to reach every station.
The key difficulty is modeling the state when switching lines. Once you pay the express entry cost, you can keep using the express line for later segments without paying it again.
Approach 1: Full Dynamic Programming with State Table (O(n) time, O(n) space)
Use dynamic programming with two states per station: the minimum cost if you arrive using the regular line and the minimum cost if you arrive using the express line. Let dp[i][0] represent reaching station i on the regular line and dp[i][1] represent reaching it on the express line.
For each station, compute transitions from the previous station. Staying on the regular line adds regular[i]. Moving to the express line either continues from express or switches from regular and adds expressCost. Iterate through the array of costs and update both states. The minimum cost to reach station i is min(dp[i][0], dp[i][1]). This approach is straightforward and clearly shows the two-line state transition.
Approach 2: Space-Optimized Dynamic Programming (O(n) time, O(1) space)
The transition only depends on the previous station, so storing the entire DP table is unnecessary. Maintain two variables: regularCost (minimum cost reaching the current station via the regular line) and expressCostState (minimum cost reaching it via the express line).
At each step, compute the new costs using the previous values. Continuing regular adds regular[i]. Continuing express adds express[i]. Switching from regular to express requires paying the one-time expressCost plus the segment cost. Update both states and record min(regularCost, expressCostState) for the answer.
This rolling-state technique removes the O(n) memory while preserving the same transition logic. The algorithm processes the cost arrays once, making it optimal for large inputs.
Recommended for interviews: The space-optimized dynamic programming approach. Interviewers expect you to identify the two-state DP (regular vs express) and then compress the state to constant memory. Writing the full DP table first shows clear reasoning; optimizing to O(1) space demonstrates stronger problem-solving skills.
We define f[i] as the minimum cost from station 0 to station i when arriving at station i by the regular route, and g[i] as the minimum cost from station 0 to station i when arriving at station i by the express route. Initially, f[0]=0, g[0]=infty.
Next, we consider how to transition the states of f[i] and g[i].
If we arrive at station i by the regular route, we can either come from station i-1 by the regular route or switch from the express route at station i-1 to the regular route. Therefore, we can get the state transition equation:
$
f[i]=min{f[i-1]+a_i, g[i-1]+a_i}
where a_i represents the cost of taking the regular route from station i-1 to station i.
If we arrive at station i by the express route, we can either switch from the regular route at station i-1 to the express route or continue on the express route from station i-1. Therefore, we can get the state transition equation:
g[i]=min{f[i-1]+expressCost+b_i, g[i-1]+b_i}
where b_i represents the cost of taking the express route from station i-1 to station i.
We denote the answer array as cost, where cost[i] represents the minimum cost from station 0 to station i. Since we can reach station i from any route, we have cost[i]=min{f[i], g[i]}.
Finally, we return cost.
The time complexity is O(n) and the space complexity is O(n), where n$ is the number of stations.
Python
Java
C++
Go
TypeScript
| Approach | Time | Space | When to Use |
|---|---|---|---|
| Dynamic Programming (2D State Table) | O(n) | O(n) | Best for understanding transitions between regular and express lines |
| Space-Optimized Dynamic Programming | O(n) | O(1) | Preferred in interviews and production when memory usage matters |
Leetcode 2361. Minimum Costs Using the Train Line: DP • Code-Yao • 1,133 views views
Watch 2 more video solutions →Practice Minimum Costs Using the Train Line with our built-in code editor and test cases.
Practice on FleetCode