HJ37 统计每个月兔子的总数

描述

有一种兔子，从出生后第3个月起每个月都生一只兔子，小兔子长到第三个月后每个月又生一只兔子。

例子：假设一只兔子第3个月出生，那么它第5个月开始会每个月生一只兔子。

一月的时候有一只兔子，假如兔子都不死，问第n个月的兔子总数为多少？

数据范围：输入满足 1≤n≤31 

输入描述：

输入一个int型整数表示第n个月

输出描述：

输出对应的兔子总数

示例1

输入：

3

输出：

2

---

#include<stdio.h>

int dp_1[31];//出生1个月的兔子数
int dp_2[31];//出生2个月的兔子数
int dp_3[31];//出生3月及以上的兔子数

int main()
{
    int n;
    scanf("%d",&n);
    int sum=0;
    for(int i=0;i<n;i++)
    {
        if (i==0) { //第1个月
            dp_3[i]=0;
            dp_1[i]=1;
            dp_2[i]=0;
        }
        else if(i==1) //第2个月
        {
            dp_3[i]=0;
            dp_1[i]=0;
            dp_2[i]=1;
        }
        else {
            dp_1[i]=dp_3[i-1]+dp_2[i-1];//第i月刚出生的兔子，即出生1月的兔子
            dp_2[i]=dp_1[i-1];//第i月出生2月的兔子
            dp_3[i]=dp_3[i-1]+dp_2[i-1];//第i月出生3月及以上的兔子
        }
    }

   sum=dp_1[n-1]+dp_2[n-1]+dp_3[n-1];
   printf("%d",sum);
}