题目地址:
https://leetcode.com/problems/minimum-costs-using-the-train-line/description/
有两条火车线路经过 0 , 1 , . . . , n 0,1,...,n 0,1,...,n这么多点,这两条线一条叫慢线,一条叫快线。 r [ i ] r[i] r[i]表示如果走慢线的话从 i → i + 1 i\rightarrow i+1 i→i+1的费用, e [ i ] e[i] e[i]表示如果走快线的话从 i → i + 1 i\rightarrow i+1 i→i+1的费用; c c c是从任意点由慢线切换到快线所需要的花费。从快线切换到慢线是不需要花费的。一个人初始在 0 0 0点,问他到各个点的最小花费。
我们把每个点拆为两个点,一个是慢线上的点,一个是快线上的点。设 f [ i ] f[i] f[i]和 g [ i ] g[i] g[i]分别是要到达慢线和快线 i i i点的最小花费。
- 对于 f [ i ] f[i] f[i],如果是从慢线 i − 1 i-1 i−1过来,则需要花 f [ i − 1 ] + r [ i − 1 ] f[i-1]+r[i-1] f[i−1]+r[i−1];如果从快线过来,则需要 g [ i − 1 ] + e [ i − 1 ] g[i-1]+e[i-1] g[i−1]+e[i−1],所以 f [ i ] = min ( f [ i − 1 ] + r [ i − 1 ] , g [ i − 1 ] + e [ i − 1 ] ) f[i] = \min(f[i - 1] + r[i - 1], g[i-1] + e[i - 1]) f[i]=min(f[i−1]+r[i−1],g[i−1]+e[i−1]);
- 对于 g [ i ] g[i] g[i]可以类似讨论。
代码如下:
class Solution {
public:
using ll = long long;
vector<ll> minimumCosts(vector<int>& reg, vector<int>& exp, int cost) {
int n = reg.size() + 1;
vector<ll> res(n - 1);
ll f[n][2];
f[0][0] = 0;
f[0][1] = cost;
for (int i = 1; i < n; i++) {
f[i][0] = min(f[i - 1][0] + reg[i - 1], f[i - 1][1] + exp[i - 1]);
f[i][1] = min(f[i - 1][0] + reg[i - 1] + cost, f[i - 1][1] + exp[i - 1]);
res[i - 1] = min(f[i][0], f[i][1]);
}
return res;
}
};
时空复杂度 O ( n ) O(n) O(n)。