算法:[动态规划] 斐波那契数列模型

目录

题目一:第 N 个泰波那契数

题目二:三步问题

题目三:最小花费爬楼梯

题目四:解码方法


题目一:第 N 个泰波那契数

泰波那契序列 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

大家都知道斐波那契数,而这道题是泰波那契数,注意是从0开始的

泰波那契数就是指,第n项是前面三项之和

所以前三项是0  1  1的话,前几个泰波那契数就是:0  1  1  2  4  7这样子的


下面采用动态规划的方法处理这道题,动态规划分为五步,前两步是最重要的:

使用动态规划时,一般会先创建一个数组,起名为dp,数组称之为dp表,把这个dp表填满,里面的某一个值可能就是最终的结果

①状态表示
②状态转移方程
③初始化
④填表顺序
⑤返回值

下面说说这五步具体的逻辑:

①状态表示

状态表示初步理解一下,就是dp表中某一个位置的值所表示的含义

状态表示是怎么确定的呢?

1、题目要求
2、经验 + 题目要求
3、分析问题的过程中,发现重复的子问题

这道题比较简单,所以就是第一种题目要求来确定状态表示,也就是:

dp[i] 表示:第 i 个泰波那契数的值

②状态转移方程

状态转移方程,也就是让 dp[i] 等于什么,而这道题已经告诉我们状态转移方程是:

dp[i] = dp[i-1] + dp[i-2] + dp[i-3]

③初始化

我们做动态规划就是为了把dp表给填满,初始化也就是保证填表的时候不越界

填表就是根据第二步得到的状态转移方程来填表,之所以要保证不越界,就拿本题中的泰波那契数的转移方程来说:
如果 i 取值为0,那代入这个状态转移方程时,就会出现dp[0] = dp[-1] + dp[-2] + dp[-3],此时这个数组会出现越界访问的问题,i 取值为1和2时,也会出现这种越界访问的问题

所以填表的时候需要保证不越界,而这道题前三道题会越界,那我们填表时,把前三个位置初始化即可,这道题目中也告诉了我们前三个值为0  1  1

④填表顺序

填表顺序就是为了填写当前状态时,所需要的状态已经计算过了

放在本题中说,也就是在填写 dp[4] 这个位置时,必须要知道dp[1]、dp[2]、dp[3]的值,所以本题的填表顺序是从左到右的,因为填当前位置时,左边的已经全部填进去了

⑤返回值

返回值就是题目要求的最终结果,即题目要求 + 状态表示,本题是求第 n 个泰波那契数,所以直接返回 dp[n] 即可


所以经过上面的五步,初步的代码是:

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

接下来可以进行空间优化,在动态规划中,空间优化一般都是使用滚动数组来进行优化的

我们发现,在代入状态转移方程时,求第 i 个位置的值时,只会用到 i 前面的三个值,其他位置的值是浪费空间的,当我们依次求 dp[i] 的时候,前面的有些状态是可以舍弃的,只需要用中间若干的状态,此时就可以用滚动数组进行优化

优化的效果就是:如果之前的空间复杂度是O(N^2),优化完就是O(N),如果之前是O(N),优化完就是O(1)

这道题中,因为我们只需要前面的三个变量,所以给出三个变量 a、b、c,当求第 d 个时,最开始是这样的:

求出来 d 等于 2 后,直接往下滚动,也就是:

又能求出最新的 d,在这个位置求完后,又能继续滚动,直到第 n 个位置,返回结果即可

下面的图能清楚的看出来滚动数组的执行过程:

也就是b的值给a,c的值给b,d的值给c,需要注意的是:
这里的赋值顺序是从前到后的,即:a = b, b = c, c = d
如果从后赋值, c = d, b = c, a = b,即每次的值都会被覆盖,c赋值给b之前,已经被d的值覆盖了,所以这种操作是不可取的

空间优化后代码:

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

提示:

        n范围在[1, 1000000]之间


假设这个小孩现在正在0号位置也就是水平地面上,面前有1、2、3、4号台阶:

0    1    2    3    4 ......

1、想上1号台阶:1种方式
①直接从0号上到1号台阶:0 -> 1

2、想上2号台阶:2种方式
①直接从0号上到2号台阶:0 -> 2
②直接从1号上到2号台阶,怎么到1号不管,但必须经过1号台阶,再到2号台阶:0 -> 1 -> 2

3、想上3号台阶:4种方式
①直接从0号上到3号台阶:0 -> 3
②直接从1号到3号,怎么到1号不管,但必须经过1号台阶:0 -> 1 -> 3
③④直接从2号到3号,怎么到2号不管,但必须经过2号台阶,因为前面的 2 中,已经说明了想上2号台阶有2种方式,所以有2种方式方式:
0 -> 2 -> 3;0 -> 1 -> 2  -> 3

4、想上4号台阶:7种方式
①直接从1号到4号,因为从0到不了,那就从1到4号台阶,怎么到1号不管,但必须经过1号台阶,而0到1的方式上面也说到了有1种方式,所以从1号到4号有1种方式:
0 -> 1 -> 4
②③直接从2号到4号,怎么到2不管,前面也说到了从0号到2号有2种方式,所以从2号到4号也有2种方式:
0 -> 2 -> 4;0 -> 1 -> 2 -> 4
④⑤⑥⑦直接从3到4,同理,怎么到3不管,前面从0到3的方式有4种,所以这里也是4种方式


所以下面根据上面所说的原理,列出dp表:

我们发现,到1、2、3号台阶的方式数量一旦直到,那么到第4号台阶的方式数量就是到1、2、3号台阶的方式数量之和,到5号台阶数量也是一样

所以继续动态规划的五步:

①状态表示

通过经验 + 题目要求,理解为:以 i 位置为结尾, ........,可以得知:

dp[i] 表示到达第 i 个台阶时,一共有多少种方法

②状态转移方程

依旧是通过经验,我们需要以 i 位置的状态,最近的一步来划分问题

由于这道题中的小孩一次能跨1阶、2阶或3阶,所以最近的一步就是从 i-1、i-2、i-3位置到 i 位置

因为从 i-3 位置到 i 位置一步就能跨到,所以只需知道从地面到 i-3 位置需要多少种方式即可,而 dp[i-3] 就是到达第 i-3 个台阶时,一共有多少种方法,所以从 i-3 位置到 i 位置就转化为了 dp[i-3] 

i-1、i-2 位置到 i 位置同理可得:dp[i-1]、dp[i-2],所以状态转移方程就是:

dp[i] = dp[i-1] + dp[i-2] + dp[i-3]

③初始化

初始化就为了填表的时候不越界,这道题是从1位置开始的,dp[0]是无意义的,所以dp[1]、dp[2]、dp[3]是会越界的,所以需要将dp[1]、dp[2]、dp[3]初始化,在上面讲述时计算了到3个台阶的方式数量,所以初始化为:

dp[1] = 1,dp[2] = 2,dp[3] = 4

④填表顺序

这道题填表顺序依旧是从左往右

⑤返回值

题目要求返回到 n 位置时,有多少种方式,而dp[n]正好就是这种意义

所以返回dp[n]即可


在初始化前,需要处理边界情况,因为n的范围是是:[1, 1000000],而我们需要初始化1、2、3,如果 n 为1,就会越界,所以需要处理边界条件

这里同样可以使用滚动数组进行优化,这里就不实现了,比较简单,和题目一中的优化几乎一摸一样,实现代码如下:

class Solution {
public:
    int waysToStep(int n) 
    {
        // 1、创建dp表
        // 2、填表前初始化
        // 3、填表
        // 4、求返回值

        //每次加法需要对MOD取模
        const int MOD = 1e9 + 7;

        if(n == 1 || n == 2) return n;
        if(n == 3) return 4;
        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;
        return dp[n] % 1000000007;
    }
};

题目三:最小花费爬楼梯

给你一个整数数组 cost ,其中 cost[i] 是从楼梯第 i 个台阶向上爬需要支付的费用。一旦你支付此费用,即可选择向上爬一个或者两个台阶。

你可以选择从下标为 0 或下标为 1 的台阶开始爬楼梯。

请你计算并返回达到楼梯顶部的最低花费。

示例 1:

输入:cost = [10,15,20]
输出:15
解释:你将从下标为 1 的台阶开始。
- 支付 15 ,向上爬两个台阶,到达楼梯顶部。
总花费为 15 。

示例 2:

输入:cost = [1,100,1,1,1,100,1,1,100,1]
输出:6
解释:你将从下标为 0 的台阶开始。
- 支付 1 ,向上爬两个台阶,到达下标为 2 的台阶。
- 支付 1 ,向上爬两个台阶,到达下标为 4 的台阶。
- 支付 1 ,向上爬两个台阶,到达下标为 6 的台阶。
- 支付 1 ,向上爬一个台阶,到达下标为 7 的台阶。
- 支付 1 ,向上爬两个台阶,到达下标为 9 的台阶。
- 支付 1 ,向上爬一个台阶,到达楼梯顶部。
总花费为 6 。

提示:

  • 2 <= cost.length <= 1000
  • 0 <= cost[i] <= 999

这道题需要注意的是,题目所给的数组的最后一个元素不是楼顶,而是台阶,楼顶是最后一个位置的下一个位置,示例一和示例二是这样子理解的:

依旧是五步:

解法一:状态表示(以 i 位置为结尾)

①状态表示

这里依旧是经验 + 题目要求

以 i 位置为结尾, .....

所以结合题目要求,在以 i 位置为结尾的前提下,这道题就理解为:到达 i 位置的最小花费

dp[i] 表示:到达 i 位置时,最小花费

②状态转移方程

状态转移方程也就是用之前或是之后的状态,推导出 dp[i] 的值

根据最近的一步,来划分问题:

到达 i 位置有两种情况:

1、先到达 i - 1 位置,然后支付 cost[i - 1] ,走一步,即:dp[i - 1] + cost[ i - 1]
2、先到达 i - 2 位置,然后支付 cost[i - 2] ,走两步,即:dp[i - 2] + cost[ i - 2]

所以得到结论:

dp[i] = min( dp[i - 1] + cost[ i - 1], dp[i - 2] + cost[ i - 2]  )

③初始化

初始化保证填表的时候不越界,因为这里需要用到 i 位置的前两个位置的cost值,所以当 i 是0或1时,会出现越界的情况,所以需要初始化

又因为题目说,可以从 0 位置或是 1 位置开始爬楼梯,所以就是说从 0 位置或是 1 位置爬楼梯不用花钱,所以都初始化为0即可

dp[0] = dp[1] = 0

④填表顺序

因为填 dp[i] 时,dp[i-1] 和 dp[i-2] 是已经算出来的,所以填表顺序也是从左往右填的

⑤返回值

想到达楼顶,所以返回值返回的是 dp[n]

所以代码如下:

class Solution 
{
public:
    int minCostClimbingStairs(vector<int>& cost) 
    {
        int n = cost.size();
        //创建dp表
        vector<int> dp(n + 1);
        //初始化这步可以不写,因为创建dp表时,默认全部初始化为0了
        dp[0] = dp[1] = 0;
        //根据状态转移方程填写dp表
        for(int i = 2; i <= n; i++)
            dp[i] = min(dp[i-1] + cost[i-1], dp[i-2] + cost[i-2]);
        //返回结果
        return dp[n];
    }
};

解法二:状态表示(以 i 位置为起点)

①状态表示

此时解法二,经验 + 题目要求,以 i 位置为起点,.......

所以结合题目要求,在以 i 位置为起点的前提下,这道题就理解为:从 i 位置到达楼顶的最小花费

dp[i] 表示:从 i 位置出发,达到楼顶的最小花费

②状态转移方程

状态转移方程也就是用之前或是之后的状态,推导出 dp[i] 的值

根据最近的一步,来划分问题:

从 i 位置出发有两种情况:

1、先支付 cost[i],然后走一步,变为从 i + 1 位置出发到达终点,即:cost[i] + dp[i + 1]
2、先支付 cost[i],然后走两步,变为从 i + 2 位置出发到达终点,即:cost[i] + dp[i + 2]

所以得到结论:

dp[i] = min( cost[i] + dp[i + 1], cost[i] + dp[i + 2] )

③初始化

初始化是为了保证填表时不越界,而这里的 dp[i] 每次填写时,需要的是后两个的cost值,所以需要先初始化后两个位置的值

因为第 n 个位置是楼顶,所以只需要初始化第 n -1 和 n - 2位置的值,所以初始化为:

dp[n - 1] = cost[n - 1],dp[n - 2] = cost[n - 2]

④填表顺序

由上面的初始化可知,我们需要知道 i 位置之后的两个位置在dp表中的值,才能知道 i 位置的值,所以填表顺序是从右向左

⑤返回值

由于 dp[i] 表示从某个位置出发,到达楼顶的最小花费,而我们刚开始是从 0 位置或是 1 位置出发的,所以需要这两个位置较小的那一个,所以返回 min(dp[0], dp[1])

代码如下:

class Solution 
{
public:
    int minCostClimbingStairs(vector<int>& cost) 
    {
        int n = cost.size();
        //创建dp表
        vector<int> dp(n);
        //初始化
        dp[n - 1] = cost[n - 1], dp[n - 2] = cost[n - 2];
        //根据状态转移方程填写dp表
        for(int i = n - 3; i >= 0; i--)
            dp[i] = min(dp[i+1] + cost[i], dp[i+2] + cost[i]);
        //返回结果
        return min(dp[0], dp[1]);
    }
};

题目四:解码方法

一条包含字母 A-Z 的消息通过以下映射进行了 编码 :

'A' -> "1"
'B' -> "2"
...
'Z' -> "26"

要 解码 已编码的消息,所有数字必须基于上述映射的方法,反向映射回字母(可能有多种方法)。例如,"11106" 可以映射为:

  • "AAJF" ,将消息分组为 (1 1 10 6)
  • "KJF" ,将消息分组为 (11 10 6)

注意,消息不能分组为  (1 11 06) ,因为 "06" 不能映射为 "F" ,这是由于 "6" 和 "06" 在映射中并不等价。

给你一个只含数字的 非空 字符串 s ,请计算并返回 解码 方法的 总数 。

题目数据保证答案肯定是一个 32 位 的整数。

示例 1:

输入:s = "12"
输出:2
解释:它可以解码为 "AB"(1 2)或者 "L"(12)。

示例 2:

输入:s = "226"
输出:3
解释:它可以解码为 "BZ" (2 26), "VF" (22 6), 或者 "BBF" (2 2 6) 。

示例 3:

输入:s = "06"
输出:0
解释:"06" 无法映射到 "F" ,因为存在前导零("6" 和 "06" 并不等价)。

提示:

  • 1 <= s.length <= 100
  • s 只包含数字,并且可能包含前导零。

这道题题意就不说明了,示例已经很清楚了,下面说动态规划实现的思路,依旧是5步:

①状态表示

同样是经验 + 状态表示:以 i 位置为结尾, ......

因为这个题目是给出一个字符串来解码,得到有多少种解码方式,所以表示到 i 位置时,有多少个解码方法数量

dp[i] 表示:以 i 位置为结尾时,解码方法的总数

②状态转移方程

根据最近的一步,划分问题

所以分为两个情况,i 位置单独解码;i 位置与 i - 1 位置结合一起解码

这两种情况都有两种可能:成功、失败

1、i 位置单独解码

解码成功:即 i 位置字符是 [1, 9] 之间的
       此时的解码数就是以 i - 1 位置为结尾时,所有的解码方案后面统一加上以 i 位置的解码成功的字符,这时的解码数量依旧是以 i - 1 位置为结尾的解码方案数,也就是dp[i - 1]
解码失败:即 i 位置字符是 0 
       此时在单独解码的方式下, 前面所有的方案都白费,因为 i 位置解码失败,整体也就失败了,所以此时的解码方案数是 0 

2、i 位置与 i - 1 位置结合一起解码

假设 i 位置字符是 a,i - 1 位置是 b

解码成功:即 a * 10 + b 是 [10, 26] 之间的,因为映射的英文字母只有26个且 a 不能为0
       此时的解码数就是以 i - 2 位置为结尾时,所有的解码方案后面统一加上以 i 位置和 i - 1 位置结合起来后的解码成功的字符,这时的解码数量依旧是以 i - 2 位置为结尾的解码方案数,也就是dp[i - 2]
解码失败:即 i 位置字符是 0 
       此时在结合解码的方式下, 前面所有的方案都白费,因为 i 位置与 i - 1 位置结合解码失败,整体也就失败了,所以此时的解码方案数是 0 

所以总结上述的描述过程,状态转移方程为:

dp[i] = dp[i - 1] + dp[i - 2](这里是解码成功时才会加)

③初始化

观察上述的状态转移方程,计算 i 位置时,需要用到 i - 1 和 i - 2 位置的值,所以需要初始化dp[0]和dp[1]的值

dp[0]:有2种情况,0/1,表示0号位置字符能否解码成功,成功就是1,失败就是0

dp[1]:有三种情况,0/1/2,表示0号和1号位置的字符,是否能够单独解码成功,或是结合解码成功,如果都不成功就是0,都成功就是2,成功一种就是1

④填表顺序

观察状态转移方程,很显然:

填表顺序是从左往右填

⑤返回值

dp[i] 表示 i 位置结尾时的解码方法总数,所以这里返回:dp[n - 1]


代码如下:

class Solution 
{
public:
    int numDecodings(string s) 
    {
        int n = s.size();
        //创建dp表
        vector<int> dp(n);
        //初始化
        dp[0] = (s[0] - '0') == 0 ? 0 : 1;
        //处理边界条件
        if(n == 1) return dp[0];

        int a = s[0] - '0';
        int b = s[1] - '0';
        if(a != 0 && b != 0) dp[1]++;
        int tmp = a * 10 + b;
        if(tmp >= 10 && tmp <= 26) dp[1]++;

        //状态转移方程,需要判断是否解码成功
        for(int i = 2; i < n; i++)
        {
            int a = s[i - 1] - '0';
            int b = s[i] - '0';
            tmp = a * 10 + b;
            //分两种情况,判断是否加编码数
            if(b != 0) dp[i] += dp[i - 1]; //单独编码时的情况
            if(tmp >= 10 && tmp <= 26) dp[i] += dp[i - 2]; //结合编码时的情况
        }
        //返回结果
        return dp[n - 1];
    }
};

下面说说细节问题,也就是处理边界问题与初始化问题的技巧:

我们可以发现上述代码编写过程中,下面两个框的内容是高度相似的

所以下面引入一种常见的方法,即:将整个dp数组向后移动一位

也就是原来的dp数组的空间是 n ,现在就变为 n + 1 了,统一移动一位后,在数组最前就多了一个0下标,也称为虚拟节点

此时我们处理边界条件时,就会比原来少处理一个位置,虽然依旧是要处理 dp[0] 和 dp[1],但是此时的dp[0]只需要根据题意赋值即可,相当于原来非常繁琐的初始化两次,减少为了一次

有两个注意事项:

①虚拟节点里面的值,需要根据题意保证后面的填表是正确的
②需要注意下标的映射关系

注意事项一中,dp[0]中的值并不是固定的,而是要根据题目条件判断,对于本道题来说,初始化完 dp[0] 和 dp[1] 后,填dp[2]时,dp[0]表示的含义就是 i 和 i - 1位置组合解码成功时需要加的值,所以如果 dp[2] 和 dp[1] 组合后解码成功,就需要加 dp[0] 的值,因此这里的 dp[0] 需要赋值为1

注意事项二中,由于dp数组统一向后移动了一位,所以在映射到原字符串s时,需要 -1 操作,对于本道题来说,dp数组向后移动后,初始化 dp[1] 时,需要判断的就是 s[1 - 1] - '0' 是否为0,而不是判断 s[1] - '0' 是否为0

优化后的代码如下:

class Solution 
{
public:
    int numDecodings(string s) 
    {
        int n = s.size();
        //创建dp表
        vector<int> dp(n + 1);
        //初始化
        dp[0] = 1;
        dp[1] = (s[1 - 1] - '0') == 0 ? 0 : 1;

        //状态转移方程,需要判断是否解码成功
        for(int i = 2; i <= n; i++)
        {
            //统一找字符串s中字符时,需要-1操作
            int a = s[i - 2] - '0';
            int b = s[i - 1] - '0';
            int tmp = a * 10 + b;
            //分两种情况,判断是否加编码数
            if(b != 0) dp[i] += dp[i - 1]; //单独编码时的情况
            if(tmp >= 10 && tmp <= 26) dp[i] += dp[i - 2]; //结合编码时的情况
        }
        //返回结果
        return dp[n];
    }
};

可以看到是比源代码优化很多的,将原代码中高度相似的代码只出现一次


动态规划——斐波那契数列模型相关的题目结束

  • 20
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值