字节跳动校招----附加题（Python）

一、题目

题目描述

存在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

输入

2
1 2

输出

4

说明

开始从房间1 只访问一次所以只能跳到p1即 房间1， 之后采用策略A跳到房间2，房间2这时访问了一次因此采用策略B跳到房间2，之后采用策略A跳到房间3，因此到达房间3需要 4 步操作。

二、分析及代码

题意：这个题挺有意思的，给了n个门，如果你是奇数次来到这个门，那么依照B规则，如果是偶数次来到这个门，则依照A规则，题目里有一个特别重要的信息是 pi(1<=pi<=i)，这说明只有偶数次来到这个门，才可以走到下一个门，奇数次都会回到当前的门或前面的门，所以这就可以用dp来解决，dp[i]表示走完i门需要移动的步数，思考一下：dp[1]是不是一定等于2，因为你第一次来到一号门，下一次肯定是移动到pi[1]号门，因为1<=pi<=i，所以pi只能取1，所以dp[1]一定等于2，那么dp[2]呢？如果pi[2]==2，那么dp[2]=dp[1]+2,想想是不是？因为第一次移动到pi[2]还是自己，第二次肯定就移动到下一个了。如果pi[2]!=2呢？那他肯定就等于dp[1] + (dp[1]-dp[0]) + 2，自己画画就可以知道，所以当pi[i]==i时，dp[i]=dp[i-1]+2，当pi[i]!=i时，dp[i] = dp[i-1] + (dp[i-1]-dp[pi[i-1]-1] + 2。代码如下：

n = int(input())
pi = [int(x) for x in input().split()]
dp = [0]*(n+1)
dp[1] = 2
k = 2     #  当前的房间号
num = 0          # 移动的总次数
while k<=n:
    if pi[k-1] == k:
        dp[k] = dp[k-1] + 2
        
    else:
        dp[k] = dp[k-1] + dp[k-1] - dp[pi[k-1]-1] + 2
    k += 1
print(dp[n]%1000000007)