进击的青蛙

试题 算法训练 进击的青蛙

资源限制

内存限制：256.0MB   C/C++时间限制：1.0s   Java时间限制：3.0s   Python时间限制：5.0s

问题描述

　　青蛙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

数据规模和约定

N <= 10^6

代码如下：

#include <iostream>
#include <string>
#define M 1000000007
using namespace std;
int a[1000000];
int main()
{
    int n;
    cin >> n;
    for (int i = 0; i < n; i++)
    {
        cin >> a[i];
    }
    if (a[0])
        a[0] = 0;
    else
        a[0] = 1;
	  if (a[1])
        a[1] = 0;
    else
        a[1] = a[0] + 1;
    if (a[2])
        a[2] = 0;
    else
        a[2] = a[0] + a[1] + 1;
    for (int i = 3; i < n; i++)
    {
        if (a[i])
            a[i] = 0;
        else
            a[i] = ((a[i - 1] + a[i - 2]) % M + a[i - 3]) % M;
    }
    if (a[n - 1])
        cout << a[n - 1] << endl;
	else
        cout << "No Way!" << endl;
    return 0;
}

运行结果截图如下：