牛客网 统计每个月兔子的总数

题目描述

有一只兔子，从出生后第3个月起每个月都生一只兔子，小兔子长到第三个月后每个月又生一只兔子，假如兔子都不死，问每个月的兔子总数为多少？

 

    /**
     * 统计出兔子总数。
     * 
     * @param monthCount 第几个月
     * @return 兔子总数
     */
    public static int getTotalCount(int monthCount)
    {
        return 0;
    }

 

 

输入描述:

输入int型表示month

输出描述:

输出兔子总数int型

示例1

输入

9

输出

34

思路：明白这是菲波那戚数列就简单了，f(n)=f(n-1)+f(n-2) （n>2）

while True:
    try:
        out = 1
        num = int(raw_input())
        if(num < 3):
            out = 1
        else:
            arr = [1,1]
            while len(arr)<num:
                arr.append(arr[-2]+arr[-1])
            out = arr[-1]
        print(out)
    except:
        break