试题 算法训练 进击的青蛙

资源限制

内存限制：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

package 算法训练;

import java.math.BigInteger;
import java.util.Scanner;

public class 进击的青蛙 {
	static long[] sum=new long[4];
	static int b=1000000007;
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner sc=new Scanner(System.in);
		int n=sc.nextInt();
		int[] a=new int[3];
		long x=0l;
		int i;
		for(i=0;i<n;i++) {
			a[i%3]=sc.nextInt();
			if(i==2) {
				if(a[0]!=1) sum[0]=1;
				if(a[1]!=1) sum[1]=sum[0]+1;
				if(a[2]!=1) sum[2]=sum[1]+sum[0]+1;
			}else if(i%3==2||i==n-1) {
				x=sum(a,n);
			}
		}
		if(x!=0) {
			System.out.print(x%b);
		}else {
            System.out.println("No Way!");
        }
	}
	public static long sum(int[] a, int n) {
        for (int i = 0; i < 3; i++) {
            if (a[i] == 1) {
                sum[i % 3] = 0;
            } else {
            	sum[i%3]=(sum[0]%b+sum[1]%b+sum[2]%b)%b;
            }
        }
 
        return sum[(n-1) % 3];
    }
}