算法训练 进击的青蛙

算法训练 进击的青蛙

问题描述
　　青蛙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

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5;
long long a[N],dp[N];
int main(){
    int n;
    cin>>n;
    dp[0]=1;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        dp[i]=0;
        if(a[i-1]==0){
            dp[i]+=dp[i-1];
        }
        if(a[i-2]==0){
            dp[i]+=dp[i-2];
        }
        if(a[i-3]==0){
            dp[i]+=dp[i-3];
        }
        dp[i]=dp[i]%1000000007;
    }
    if(dp[n]==0||a[n]!=0){
        cout<<"No Way!";
    }else{
        cout<<dp[n];
    }
}