大厂Java笔试题之统计兔子出生问题

题目：有一种兔子，从出生后第3个月起每个月都生一只兔子，小兔子长到第三个月后每个月又生一只兔子。  例子：假设一只兔子第3个月出生，那么它第5个月开始会每个月生一只兔子。  一月的时候有一只兔子，假如兔子都不死，问第n个月的兔子总数为多少？  数据范围：输入满足  1≤n≤31  输入描述： 输入一个int型整数表示第n个月  输出描述：  输出对应的兔子总数

 例如，输入3，输出2 

看到这个题目，首先先按题意列出前几年的情况，得到如下数据：

public class Demo11 {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        // 注意 hasNext 和 hasNextLine 的区别
        while (in.hasNextInt()) { // 注意 while 处理多个 case
            int a = in.nextInt();
            System.out.println(countRabbits(a));
        }
    }

    public static int countRabbits(int n) {
        if (n < 1 || n > 31) {
            throw new IllegalArgumentException("Input value must be between 1 and 31 inclusive.");
        }

        // 初始两个月兔子数量分别为1和1
        int previous = 1;
        int current = 1;

        // 从第三个月开始计算兔子总数
        for (int month = 3; month <= n; month++) {
            // 计算下一个月的兔子总数为前两个月之和
            int next = previous + current;

            // 更新前两个月的兔子数量
            previous = current;
            current = next;
        }

        // 返回第n个月的兔子总数
        return current;
    }
}

  如果大家需要视频版本的讲解，欢迎关注我的B站：