算法题-n月后兔子数量

public class test3 {
    public static void main(String args[]){
        Scanner sc = new Scanner(System.in);
        while(sc.hasNext()) {
            int n = sc.nextInt();
            int num = fun(n);
            System.out.println(num);
        }
    }

    public static int fun(int month){
        if(month < 5){
            return 1;
        }else{
            int num = 0;
            int a1 = 1;
            int a2 = 0;
            int a3 = 0;
            int a4 = 0;
            for(int i=2; i<=month; i++){
                num += a4;
                a4 = a3;
                a3 = a2;
                a2 = a1;
                a1 = num;
            }
            return num + a1 + a2 + a3 + a4;
        }
    }
}