【Leetcode】1997. First Day Where You Have Been in All the Rooms

题目地址：

https://leetcode.com/problems/first-day-where-you-have-been-in-all-the-rooms/description/

有$n$个房间，编号$0\sim n-1$。一开始你在$0$号房间，再给定一个长$n$数组$ne$。如果你是第奇数次访问$i$号房间，那么下一步你会访问$ne[i]$号房间，题目保证$0\le ne[i]\le i$；如果是偶数次，则下一步会访问$i+1$号房间。问访问完所有房间至少需要多少多少步。可以证明所有房间都会访问完。答案模$10^9+7$返回。

因为每一步要么是向右走$1$步，要么是回头，所以容易知道如果要从$a$走到$b$，中间的房间一定都要走过。并且，第一次访问$k$的时候一定是从$k-1$走过来的。设$f[k]$是第一次访问$k$要多少步。我们考虑从$0$到$k$的整个流程，这个流程一定是：$$0\Rightarrow ne[k-1]\Rightarrow ne[k-1]\Rightarrow k-1\Rightarrow ne[k-1]\Rightarrow ne[k-1]\Rightarrow k-1\Rightarrow k$$其中：

1. $0\Rightarrow ne[k-1]\Rightarrow ne[k-1]\Rightarrow k-1$要花$f[k-1]$步；
2. $k-1\Rightarrow ne[k-1]$要花$1$步；
3. $ne[k-1]\Rightarrow ne[k-1]\Rightarrow k-1$要花$f[k-1]-f[ne[k-1]]$步（因为这是第奇数次走到$ne[k-1]$的情况）
4. $k-1\Rightarrow k$要花$1$步。

加起来即可。代码如下：

class Solution {
 public:
  int firstDayBeenInAllRooms(vector<int>& ne) {
    constexpr int MOD = 1e9 + 7;
    int n = ne.size();
    vector<int> f(n);
    f[0] = 0;
    for (int i = 1; i < n; i++)
      f[i] = ((f[i - 1] + 1 + (f[i - 1] - f[ne[i - 1]]) + 1) % MOD + MOD) % MOD;

    return f[n - 1];
  }
};

时空复杂度$O(n)$。