字节跳动校招----附加题(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)

 

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值