java 斐波那契数列考虑死亡的实现

题目描述

母牛从3-7岁初每年会生产一头小母牛，10岁后死亡（10岁仍然存活）。假设初始有1头刚出生的母牛，请问第n年有多少头母牛？
实例 输入 2 输出 1
输入3 输出 2
输入12 输出123

package com.bytedance;

/**
 * @author ZFX
 * @date2019/5/4 9:58.
 */
import java.util.*;
public class Cow {
    public static void main(String[] args){
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int count = res(n);
        System.out.println(count);
    }
    public static int res(int n){
        int[] cowsOfYears = new int[10];
        cowsOfYears[0] = 1;
        for(int i = 1; i < n; i++){
            //使用更新删除死亡的牛
            for(int j = 9; j > 0; j--){
                cowsOfYears[j] = cowsOfYears[j - 1];
            }
            //刚出生的牛的数目
            int tmp = cowsOfYears[2] + cowsOfYears[3] + cowsOfYears[4] + cowsOfYears[5] + cowsOfYears[6];
            cowsOfYears[0] = tmp;
        }
        int count = 0;
        //计算牛的总数目
        for(int i = 0; i < 10; i++){
            count +=cowsOfYears[i];
        }
        return count;
    }
}