蓝桥杯训练-母牛的故事

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

输入

输入数据由多个测试实例组成，每个测试实例占一行，包括一个整数n(0<n<55)，n的含义如题目中描述。
n=0表示输入数据的结束，不做处理。

输出

对于每个测试实例，输出在第n年的时候母牛的数量。
每个输出占一行。

样例输入

2
4
5
0

样例输出

2
4
6

java通过代码

import java.util.Scanner;
public class Main{
    public static void main(String[] args)
    {
        int[] a = new int[55];
        int count = 0;
        Scanner scan = new Scanner(System.in);
        while(true)
        {
            int n  = scan.nextInt();
            if(n==0) break;
            else {
                a[count++] = n;
            }
        }
        int[] arry = new int[55];
        arry[0] = 0;
        arry[1] = 1;
        arry[2] = 2;
        arry[3] = 3;
        for(int j=4;j<55;j++)
        {
            arry[j] = arry[j-3]+arry[j-1];    
        }    
        for(int i=0;i<count;i++)
        {
            int m = a[i];
            System.out.println(arry[m]);
        }
    }
}