跳房子（字节算法题）

海带卷

已于 2022-07-14 12:19:54 修改

阅读量575

点赞数

文章标签：算法 java

于 2022-07-14 12:16:35 首次发布

本文链接：https://blog.csdn.net/qq_40081651/article/details/125780976

版权

这是一篇关于字节跳动算法题“跳房子”的解题报告。文章介绍了问题背景，提供了动态规划的解题思路，以及具体的Java代码实现。在测试过程中，作者注意到在模运算后可能出现dp[i]小于dp[p[i]]的情况，并对此进行了分析。

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

存在n+1个房间，每个房间依次为房间1 2
3…i，每个房间都存在一个传送门，i房间的传送门可以把人传送到房间pi(1<=pi<=i),现在路人甲从房间1开始出发(当前房间1即第一次访问)，每次移动他有两种移动策略：
A. 如果访问过当前房间 i 偶数次，那么下一次移动到房间i+1；
B. 如果访问过当前房间 i 奇数次，那么移动到房间pi；现在路人甲想知道移动到房间n+1一共需要多少次移动；

输入描述: 第一行包括一个数字n(30%数据1<=n<=100，100%数据 1<=n<=1000)，表示房间的数量，接下来一行存在n个数字
pi(1<=pi<=i), pi表示从房间i可以传送到房间pi。

输出描述: 输出一行数字，表示最终移动的次数，最终结果需要对1000000007 (10e9 + 7) 取模。
示例1
输入
3
1 2 2

输出
8

思路：动态规划

在这里插入图片描述
递推：
dp[i]表示走到第i个房子前经历的步数
dp[1] = 0
dp[2] =2
dp[3] =4
dp[4] = dp[3]+1 +(dp[3]-dp[2])+1

找规律：i从1开始
dp[1] =0
当p[i] == i 时 dp[i] =dp[i-1]+2;
当p[i] != i 时 dp[i] = dp[i-1] +1+(dp[i-1]-dp[p[i-1]])+1=2*dp[i-1] - dp[p[i-1]] +2;

再进行合并：
dp[1] =0
当i>1时： dp[i] =2*dp[i-1] - dp[p[i-1]] +2;

题解（java）

import java.util.Scanner;

public class Main {
   
    public static void main(String[] args) {
   
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int[] p = new int[n+1];
        long[] dp = new long[n+2];
        dp[1]=0;
        p[1] = in.nextInt();
        long mod = (long)(1e9+7);
        for(int i=2;i<=n;i++){
   
            p[i] = in.nextInt();
            dp[i] = (2*dp[i-1]-dp[p[i-1]]+2) % mod;
        }
        dp[n+1