动态规划--访问完所有房间的第一天

牛顿没有错

已于 2024-03-29 15:40:47 修改

阅读量827

点赞数 16

分类专栏：算法题文章标签：动态规划算法 leetcode java

于 2024-03-28 22:53:45 首次发布

本文链接：https://blog.csdn.net/qq_54481628/article/details/137126724

版权

算法题专栏收录该内容

23 篇文章 0 订阅

订阅专栏

文章描述了一个关于房间访问的问题，根据给定的nextVisit数组，按照特定的奇偶性规则决定每天的房间顺序。作者提出了使用动态规划方法求解，计算从0号房间出发，访问所有房间所需的最早日期。解决方案涉及转移方程和前缀和优化。

摘要由CSDN通过智能技术生成

题目描述

你需要访问 n 个房间，房间从 0 到 n - 1 编号。同时，每一天都有一个日期编号，从 0 开始，依天数递增。你每天都会访问一个房间。

最开始的第 0 天，你访问 0 号房间。给你一个长度为 n 且 下标从 0 开始 的数组 nextVisit 。在接下来的几天中，你访问房间的次序将根据下面的规则决定：

假设某一天，你访问 i 号房间。
如果算上本次访问，访问 i 号房间的次数为奇数，那么 第二天 需要访问 nextVisit[i] 所指定的房间，其中 0 <= nextVisit[i] <= i 。
如果算上本次访问，访问 i 号房间的次数为偶数，那么 第二天 需要访问 (i + 1) mod n 号房间。

请返回你访问完所有房间的第一天的日期编号。题目数据保证总是存在这样的一天。由于答案可能很大，返回对 10^9 + 7 取余后的结果。

示例 1：

输入：nextVisit = [0,0]
输出：2
解释：
- 第 0 天，你访问房间 0 。访问 0 号房间的总次数为 1 ，次数为奇数。
  下一天你需要访问房间的编号是 nextVisit[0] = 0
- 第 1 天，你访问房间 0 。访问 0 号房间的总次数为 2 ，次数为偶数。
  下一天你需要访问房间的编号是 (0 + 1) mod 2 = 1
- 第 2 天，你访问房间 1 。这是你第一次完成访问所有房间的那天。

示例 2：

输入：nextVisit = [0,0,2]
输出：6
解释：
你每天访问房间的次序是 [0,0,1,0,0,1,2,...] 。
第 6 天是你访问完所有房间的第一天。

示例 3：

输入：nextVisit = [0,1,2,0]
输出：6
解释：
你每天访问房间的次序是 [0,0,1,1,2,2,3,...] 。
第 6 天是你访问完所有房间的第一天。

提示：

n == nextVisit.length
2 <= n <= 10^5
0 <= nextVisit[i] <= i

1. 思路

分析题意发现 nextVisit[i] 的范围属于 [0,i]，意味着当你首次到达房间 i 时会回退到房间 nextVisit[i]。而只有访问过该房间偶数次时才会到达下一个房间，进而推断出到达 i 时，[0,i) 的房间已经被访问过偶数次。

定义 f[i] 表示从奇数次到房间 iii，到奇数次到达房间 i+1 所需要的天数。以下用 to 代表 nextVisit[i]，回退到房间 to 时是奇数次访问，又需要花费 f[to] 才会到达房间 to+1。从 i 访问 to 和 i+1 又分别需要花费一天，所以有转移方程:

$f[i]=\sum_{j=to}^{i-1}f[j]+2$

但这样做时间复杂度是 O(n^2)，考虑到求和可以用前缀和预处理达到 O(1) 转移。定义 dp[i] 表示从 0 号房间出发首次到 i+1 花费的天数，经化简有转移方程:

$dp[i]=dp[i-1]-dp[to]+2$

dp[n−1] 就是第一次到达 n 号房间花费的天数。

2. 代码

class Solution {
    public int firstDayBeenInAllRooms(int[] nextVisit) {
        int mod = 1000000007;
        int len = nextVisit.length;
        int[] dp = new int[len];

        dp[0] = 2; //初始化原地待一天 + 访问下一个房间一天
        for (int i = 1; i < len; i++) {
            int to = nextVisit[i];
            dp[i] = 2 + dp[i - 1];
            if (to != 0) {
                dp[i] = (dp[i] - dp[to - 1] + mod) % mod; //避免负数
            }

            dp[i] = (dp[i] + dp[i - 1]) % mod;
        }
        return dp[len - 2]; //题目保证n >= 2
    }
}