试题 算法训练 进击的青蛙

资源限制

时间限制：1.0s   内存限制：256.0MB

问题描述

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

思路：递推，数组记录到达i点的方法数

#include<iostream>
using namespace std;

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