【Leetcode】2361. Minimum Costs Using the Train Line

题目地址：

https://leetcode.com/problems/minimum-costs-using-the-train-line/description/

有两条火车线路经过$0,1,...,n$这么多点，这两条线一条叫慢线，一条叫快线。$r[i]$表示如果走慢线的话从$i\to i+1$的费用，$e[i]$表示如果走快线的话从$i\to i+1$的费用；$c$是从任意点由慢线切换到快线所需要的花费。从快线切换到慢线是不需要花费的。一个人初始在$0$点，问他到各个点的最小花费。

我们把每个点拆为两个点，一个是慢线上的点，一个是快线上的点。设$f[i]$和$g[i]$分别是要到达慢线和快线$i$点的最小花费。

1. 对于$f[i]$，如果是从慢线$i-1$过来，则需要花$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])$；
2. 对于$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)$。