经典动态规划思想：dp与滚动数组

题目

泰波那契序列 Tn 定义如下： 

T0 = 0, T1 = 1, T2 = 1, 且在 n >= 0 的条件下 Tn+3 = Tn + Tn+1 + Tn+2

给你整数 n，请返回第 n 个泰波那契数 Tn 的值。

示例 1：

输入：n = 4
输出：4
解释：
T_3 = 0 + 1 + 1 = 2
T_4 = 1 + 1 + 2 = 4

示例 2：

输入：n = 25
输出：1389537

提示：

• 0 <= n <= 37
• 答案保证是一个 32 位整数，即 answer <= 2^31 - 1。

代码实现

1.创建dp表

class Solution {
public:
    int tribonacci(int n) 
    {
        if(n==0) return 0;//处理边界情况
        if(n==1||n==2) return 1;
        vector<int> dp(n+1);
        dp[0]=0;dp[1]=1,dp[2]=1;//初始化数据
        for(int i=3;i<=n;i++)//创建dp表
        {
            dp[i]=dp[i-1]+dp[i-2]+dp[i-3];
        }
        return dp[n];
    }
};

2.滚动数组

class Solution {
public:
    int tribonacci(int n) 
    {
        if(n==0) return 0;
        if(n==1||n==2) return 1;
        int a=0,b=1,c=1,d=0;
        for(int i=3;i<=n;i++)
        {
            d=a+b+c;
            a=b;b=c;c=d;
        }
        return d;
    }
};

题目

三步问题。有个小孩正在上楼梯，楼梯有n阶台阶，小孩一次可以上1阶、2阶或3阶。实现一种方法，计算小孩有多少种上楼梯的方式。结果可能很大，你需要对结果模1000000007。

示例1:

 输入：n = 3 
 输出：4
 说明: 有四种走法

示例2:

 输入：n = 5
 输出：13

提示:

1. n范围在[1, 1000000]之间

代码实现

1.创建dp表

class Solution {
public:
    int waysToStep(int n) 
    {
        if(n==1) return 1;
        if(n==2) return 2;
        if(n==3) return 4;
        const int MOD=1e9+7;
        vector<int> dp(n+1);
        dp[1]=1;dp[2]=2;dp[3]=4;//先初始化前几位数据
        for(int i=4;i<=n;i++)
        {
            dp[i]=((dp[i-1]+dp[i-2])%MOD+dp[i-3])%MOD;//创建dp表
        }
        return dp[n];
    }
};

2.滚动数组

class Solution {
public:
    int waysToStep(int n) 
    {
        if(n==1) return 1;
        if(n==2) return 2;
        if(n==3) return 4;
        const int MOD=1e9+7;
        int a=1,b=2,c=4,d=0;//初始化数据
        
        for(int i=4;i<=n;i++)
        {
            d=((a+b)%MOD+c)%MOD;//滚动的朝n走
            a=b;
            b=c;
            c=d;
        }
        return d;
    }
};