母牛的故事-Java

题目描述

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

输入

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

输出

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

样例输入

2
4
5
0

样例输出

2
4
6

参考解答

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int num;
		num = sc.nextInt();
		
		while (num != 0) {
			System.out.println(count(num));
			num = sc.nextInt();
		};
	}
	
	private static int count(int n) {
		if (n < 5) {
			return n;
		}
		else {
			int temp[] = new int[n + 1];
			temp[1] = 1;
			temp[2] = 2;
			temp[3] = 3;
			temp[4] = 4;
			for (int i = 5; i <= n; i++) {
				temp[i] = temp[i - 1] + temp[i - 3];
			}
			return temp[n];
		}
	}
}