试题 算法训练 进击的青蛙

问题描述

　　青蛙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>
using namespace std;

long long load[1000000] = {0};
int N;

int main() {
	cin >> N;
	
	for (int i = 1; i <= N; i ++) {
		cin >> load[i];
	}
	
	if (load[1] == 0) {
		load[1] = 1;
	} else {
		load[1] = 0;
	}
	
	if (load[2] == 0) {
		load[2] = load[1] + 1;
	} else {
		load[2] = 0;
	}
	
	if (load[3] == 0) {
		load[3] = load[1] + load[2] + 1;
	} else {
		load[3] = 0;
	}
	
	for (int i = 4; i <= N; i ++) {
		if (load[i] == 1) {
			load[i] = 0;
		} else {
			load[i] = (load[i - 1] % 1000000007 + load[i - 2] % 1000000007 + load[i - 3] % 1000000007) % 1000000007;
		}
	}
	
	if (load[N] > 0) {
		cout << load[N];
	} else {
		cout << "No Way!";
	}
	
	return 0;
}

总结：

类似斐波那契数列

只不过这里是三个三个的加

一开始用的函数递归

结果全部超时

只能换一种思路