一、题目
题目描述
存在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)