蓝桥杯-进击的青蛙

问题描述

　　青蛙X正准备跳过一座桥，这座桥被划分为N段，记青蛙所在的起始点为0，桥的末端为N。桥上的一些点有一些石子，这些点是无法跳上去的。青蛙每次跳跃能向前跳跃+1，+2，+3段，现在请你算出跳到末端的总方法数。如果无法到达，请输出”No Way!"

输入格式

　　输入数据共N行。

　　第一行一个数字N，代表桥的长度。

　　接下来N行，表示从点1~N的道路情况，每行一个数字0或1，1表示有石子。

输出格式

　　输出一行，为一个整数，代表方法数，无法到达为“No Way!"

　　由于数据过大，我们只需要求出 对 1000000007 的余数即可

样例输入

5

1

0

0

1

0

样例输出

3

思路：

初步一看，无法用搜索类算法解决，所以考虑动态规划思想，发现此题很想斐波那契数列：

相当经典的 “斐波那锲数列” 的变形题目了！，斐波那锲数列：当项数大于二时，f[i] = f[i-1] + f[i-2]。

这个题目就是当项数大于三时，dp[i] = dp[i-1] + dp[i-2] + dp[i-3];

然后呢，当数字是1的时候，就表示是有石子，青蛙不能跳上去，dp[i] = 0;

初始值呢？

就是 i == 1,2,3的时候，当i == 1&& 数字!=1 的时候，青蛙只能跳一步跳上去，dp[1] = 1;

当i == 2 && 数字 != 1的时候，青蛙可以直接跳两步跳上去，也可以从1跳一步跳上去，dp[2] = dp[1] + 1;

当i == 3 && 数字 != 1的时候，青蛙可以直接跳三步跳上去，也可以从1跳两步跳上去，也可以从2跳一步跳上去，

dp[3] = dp[2] +dp[1] + 1;

n=int(input())
nums=[]
for i in range(n):
    nums.append(int(input()))
dp=[0 for i in range(n+1)]
dp[0]=0
if nums[0]!=1:
    dp[1]=1
else:
    dp[1]=0
if nums[1]!=1:
    dp[2]=dp[1]+1
else:
    dp[2]=0
if nums[2]!=1:
    dp[3]=dp[2]+dp[1]+1
else:
    dp[3]=0
for i in range(4,n+1):
    if nums[i-1]==1:
        dp[i]=0
    if nums[i-1]!=1:
        dp[i]=(dp[i-1]+dp[i-2]+dp[i-3])%1000000007
if dp[n]==0:
    print("No Way!")
else:
    print(dp[n])