母牛的故事

母牛的故事

题目描述：
有一头母牛，它每年年初生一头小母牛。每头小母牛从第四个年头开始，每年年初也生一头小母牛。请编程实现在第n年的时候，共有多少头母牛？

这道题的关键点在每年增长的牛的数目，从这个列表中我们不难发现从第5年后牛的每年增长数为3年前的牛的总数。那么牛的总数为前一年的牛的总数加上三年前牛的总数。
代码：

import java.util.Scanner;
public class zs {
	public static void main(String[] args) {
		int[] arr=new int[55];//arr数组存放牛数
		int[] y =new int[55];//存放输入的年数
		Scanner scanner = new Scanner(System.in);
		for(int i=0;i<y.length;i++){
			 y[i]=scanner.nextInt();
			if(y[i]==0)
				break;
		}
		 xiaoniushu(y,arr);
	}
	public static void xiaoniushu(int y[],int arr[]){
		//z是本年长大的牛，n是已经长大的牛,arr存放年增长牛数
		
			for( int k=1;k<y.length;k++){
				if(k>4){
					arr[k]=arr[k-1]+arr[k-3];//今年的牛数等于去年的牛数加上今年新增的牛数（新增牛数等于三年前牛数）
				}
				else arr[k]=arr[k-1]+1;
			}
			for(int j=0;j<y.length;j++){
				if(y[j]==0)break;
				int d =y[j];
				int niu=arr[d];
				System.out.println(niu);
			}
	}
}

运行结果：
1,输入年数：

2,输出结果