7-245 斐波那契数列

意大利数学家斐波那契（Leonardo Fibonacci）是12、13世纪欧洲数学界的代表人物。他提出的“兔子问题”引起了后人的极大兴趣。
“兔子问题”假定一对大兔子每一个月可以生一对小兔子，而小兔子出生后两个月就有繁殖能力，问从一对小兔子开始，n个月后能繁殖成多少对兔子？

输入格式:

首先输入一个正整数T，表示测试数据的组数，然后是T组测试数据。每组测试输入一个正整数n（1≤n≤46）。

输出格式:

对于每组测试，输出n个月后能繁殖成的兔子对数。

输入样例:

1
4

输出样例:

3

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

思路

假设 n 个月后兔子的数目为f(n)，则当n=1或n=2时，兔子总数均为1 当n≥3时，第n月的兔子总数 = 第n-1月的兔子总数 +　第n-2月的兔子总数

根据上述公式，我们可以写出以下递归代码：

int fib(int n)
{
    if (n == 1 || n == 2) {         // 如果是第1个月或第2个月，返回1
        return 1;
    } else {
        return fib(n - 1) + fib(n - 2);  // 否则返回前两个月的兔子总数之和
    }
}

但使用上述代码，PTA会超时。为了避免这种问题，在计算过程中，可以用一个缓存数组 cache 来存储已经计算出来的斐波那契数列的值，从而避免重复计算。

#define MAXN 50             // 最大的斐波那契数列长度
int cache[MAXN];            // 用来保存已经计算得到的结果

// 计算第 n 个斐波那契数
int fib(int n)
{
    if (n == 1 || n == 2) {     // 如果是第1个月或第2个月，返回1
        return 1;
    }
    if (cache[n] != 0) {        // 如果之前已经计算过，则从缓存中取出结果
        return cache[n];
    }
    int result = fib(n - 1) + fib(n - 2);  // 否则递归计算 f(n)，并将结果保存到缓存中
    cache[n] = result;
    return result;
}

在main函数中，首先读入需要测试的数据组数 T。接下来，每次进入while循环，读入当前测试数据中兔子的繁殖时间n，并调用函数 fib(n) 求得兔子的繁殖对数，最后将结果输出。

在fib函数中，首先判断当前月份n是否为1或2，若是则直接返回1；否则，如果之前已经计算过第n个斐波那契数，则直接从缓存数组cache中取值；如果没有，则采用递归方式分别计算第n-1个和n-2个斐波那契数，再将结果相加并将结果保存到缓存数组cache中，避免重复计算。最后，返回计算得到的第n个斐波那契数。

代码

#include <stdio.h>

#define MAXN 50             // 最大的斐波那契数列长度
int cache[MAXN];            // 用来保存已经计算得到的结果

// 计算第 n 个斐波那契数
int fib(int n)
{
    if (n == 1 || n == 2) {     // 如果是第1个月或第2个月，返回1
        return 1;
    }
    if (cache[n] != 0) {        // 如果之前已经计算过，则从缓存中取出结果
        return cache[n];
    }
    int result = fib(n - 1) + fib(n - 2);  // 否则递归计算 f(n)，并将结果保存到缓存中
    cache[n] = result;
    return result;
}

int main()
{
    int T, n;
    scanf("%d", &T);           // 输入测试数据组数
    while (T--) {
        scanf("%d", &n);       // 输入兔子繁殖时间
        printf("%d\n", fib(n));  // 输出兔子对数
    }
    return 0;
}