Leetcode-1997-访问完所有房间的第一天-c++

题目详见https://leetcode.cn/problems/first-day-where-you-have-been-in-all-the-rooms/

t o = n e x t v i s i t $i$ : 第一次到 i 之后回退到的位置 to=nextvisit $i$ : 第一次到i之后回退到的位置 to=nextvisit $i$ :第一次到i之后回退到的位置
f $i$ : 从奇数次到 i 到奇数次到 i + 1 所需要的天数 f $i$ : 从奇数次到 i 到奇数次到i+1所需要的天数 f $i$ :从奇数次到i到奇数次到i+1所需要的天数
d p $i$ : 从 0 号房间出发到 i + 1 号房间所需要的天数 dp $i$ : 从0号房间出发到i+1号房间所需要的天数 dp $i$ :从0号房间出发到i+1号房间所需要的天数

首次到第i+1个房间 d p $i$ dp $i$ dp $i$ 分4步：

(奇数次到 i i i)从起点走到 i i i； T = d p $i - 1$ T=dp $i-1$ T=dp $i-1$

从 i i i跳到 n e x t V i s i t $i$ nextVisit $i$ nextVisit $i$ ； T = 1 T=1 T=1

从 n e x t V i s i t $i - 1$ nextVisit $i-1$ nextVisit $i-1$ 再走到 i i i； T = d p $i - 1$ − d p $n e x t v i s i t \[ i − 1$ ] T=dp $i-1$ -dp $nextvisit\[i-1$ ] T=dp $i-1$ −dp $nextvisit\[i−1$ ]

(偶数次到 i i i)从 i i i走到 i + 1 i+1 i+1； T = 1 T=1 T=1

综上
d p $i$ = d p $i - 1$ + 1 + d p $i - 1$ − d p $n e x t v i s i t \[ i − 1$ ] + 1 dp $i$ =dp $i-1$ + 1 + dp $i-1$ -dp $nextvisit\[i-1$ ] + 1 dp $i$ =dp $i-1$ +1+dp $i-1$ −dp $nextvisit\[i−1$ ]+1

对你没看错是题解错了

注释代码

cpp 复制代码

class Solution {
public:
    int firstDayBeenInAllRooms(vector<int>& nextVisit) {
        int mod = 1e9 + 7;
        int len = nextVisit.size();
        vector<int> dp(len);

        dp[0] = 2; //初始化原地待一天 + 访问下一个房间一天（第二步的时间）
        for (int i = 1; i < len; i++) {
            int to = nextVisit[i];
            dp[i] = 2 + dp[i - 1];	//1，2，4步
            
            if (to != 0) {
            	// 这里是第三步的第一部分，我做的时候主要这里这个dp[to - 1]比较难理解
            	// dp[to] 代表的是从0到to+1，所以想到to的话就是dp[to - 1]
                dp[i] = (dp[i] - dp[to - 1] + mod) % mod; //避免负数
            }
            dp[i] = (dp[i] + dp[i - 1]) % mod; // 这里是第三步的第二部分
        }
        return dp[len - 2]; //输出dp[n-2], 也就是到n-1的距离，而此时恰好下标从0开始。
    }
};

笔者也在新手学习期中，所写的内容主要与大家交流学习使用，如有发现任何问题敬请指正！