动态规划解题方法

代码如下:

class Solution {

    public int minimumTotal(List<List<Integer>> triangle) {

      // 创建表格

        int[][] dp = new int[triangle.size()][triangle.size()];

        dp[0][0] = triangle.get(0).get(0);

      // 动态规划的方程式

        for (int i = 1; i < dp.length; i++) {

            for (int j = 0; j <= i; j++) {

                int curr = triangle.get(i).get(j);

                if (j == 0) {

                    dp[i][j] = dp[i-1][j] + curr;

                } else if (j == i) {

                    dp[i][j] = dp[i-1][j-1] + curr;

                } else {

                    dp[i][j] = Math.min(dp[i-1][j-1], dp[i-1][j]) + curr;

                }

            }

        }

        int res = dp[triangle.size()-1][0];

        for (int j = 1; j < triangle.size(); j++) {

            res = Math.min(res, dp[triangle.size()-1][j]);

        }

        return res;

    }

}

下面来分析这个问题的时间复杂度以及空间复杂度,一般来说空间复杂度是就是 DP 表格的大小。

在这道问题中是 O(n^2),而对于时间复杂度来说,就是整个 dp 的遍历次数,而在这个问题中我们只进行了一次遍历,也即一个矩阵的遍历,所以是O(n^2)。

而如果想要优化到 O(n),我们需要怎么做呢?

实际上这个就涉及到了一种状态压缩的方法,也即压缩这个状态表。

那么怎么去压缩呢?

这个问题比较简单,因为dp[i][j]仅仅与上一层的状态有关,所以说与前两层的是没有任何关系的,因此我们不必存储这些无关的状态。

实际上最简单的状态压缩就是保留好前两个状态即可,例如在计算第四行的时候,保留第三行以及第二行的状态表,然后交替的进行更新就可以啦。

class Solution {

    public int minimumTotal(List<List<Integer>> triangle) {

      // 只保留最近 2 行

        int[][] dp = new int[2][triangle.size()];

        dp[0][0] = triangle.get(0).get(0);

        for (int i = 1; i < triangle.size(); i++) {

            int row = i % 2;

            int prevRow = (i-1) % 2;

            for (int j = 0; j <= i; j++) {

                if (j == 0) {

                    dp[row][j] = dp[prevRow][j] + triangle.get(i).get(j);

                } else if (j == i) {

                    dp[row][j] = dp[prevRow][j-1] + triangle.get(i).get(j);

                } else {

                    dp[row][j] = Math.min(dp[prevRow][j-1], dp[prevRow][j]) + triangle.get(i).get(j);

                }

            }

        }

        int res = dp[(triangle.size() - 1) % 2][0];

        for (int j = 1; j < triangle.size(); j++) {

            res = Math.min(res, dp[(triangle.size() - 1) % 2][j]);

        }

        return res;

    }

}

这个的空间复杂度是 O(2n),能不能压缩成严格意义上的O(n)呢?

那么再往后是否还能够进行状态的压缩呢?

答案是可以的,我们可以再想一种方程然后达到最优的空间复杂度的目标。

当我们在计算位置 [i][j] 时,f[j+1] 到 f[i] 已经是第 i 行的值,而 f[0] 到 f[j] 仍然是第 i-1 行的值。

此时我们直接通过 f[j] = min(f[j−1], f[j]) + c[i][j]来计算,但是这个时候我们需要j 是倒着遍历的,因为这样才不会影响之前记录下的状态。

如果从 1 开始,那么计算 2 的时候就会用到新的 1 的数值而不是上一层 1 的数值。

代码如下:

class Solution {

    public int minimumTotal(List<List<Integer>> triangle) {

        int[] dp = new int[triangle.size()];

        dp[0] = triangle.get(0).get(0);

        for (int i = 1; i < triangle.size(); i++) {

            dp[i] = dp[i-1] + triangle.get(i).get(i);

            for (int j = i-1; j > 0; j--) {

                dp[j] = Math.min(dp[j-1], dp[j]) + triangle.get(i).get(j);

            }

            dp[0] += triangle.get(i).get(0);

        }

        int res = dp[0];

        for (int j = 1; j < triangle.size(); j++) {

            res = Math.min(res, dp[j]);

        }

        return res;

    }

}

方法 2

方法 1 有点绕,但如果自下向上来理解,就会变得很简单,这个方法也叫 bottom-up,方法 1 则是 top-down。

从结果出发,这个问题是一个发散三角树的问题,从最后一行出发,然后每一行每一行的进行递推,那么第一行就是最终的结果了。

举个最简单的例子:

1

1 2

如果从最底下往上出发,实际上找最小值方法的规律很容易找到,那就是在第二行[1, 2]里面选择一个就可以了,因为他们两个都要走到根节点。

也就是在下一行的两个数里面取个小的就行了,那么结果就是第一行的数值。

我们先用二维的转移矩阵来解释这个问题,用这种方法也不需要考虑方法 1 里面的边界条件了:

dp[i][j] = min(dp[i+1][j], dp[i+1][j+1]) + c[i][j]

状态转换图如下所示:

class Solution {

    public int minimumTotal(List<List<Integer>> triangle) {

        int[][] dp = new int[triangle.size()][triangle.size()];

        // 创建 DP 空间

        for (int i = 0; i < triangle.size(); i++) {

            dp[triangle.size() - 1][i] = triangle.get(triangle.size() - 1).get(i);

        }

        for (int i = triangle.size() - 2; i >= 0; i--) {

            for (int j = 0; j < triangle.get(i).size(); j++) {

                dp[i][j] = Math.min(dp[i+1][j], dp[i+1][j+1]) + triangle.get(i).get(j);

            }

        }

        return dp[0][0];

    }

}

那么在进行状态压缩的时候,我们该怎么去做呢?

实际上就是只用一个状态表来表示所有的。

因为只是和上一个状态相关,所以说可以表示成如下的形式:

dp[j] = min(dp[j], dp[j + 1]) + triangle[i][j],

我们只用 j 来代表当前的状态,然后最终输出dp[0]即可。

class Solution {

    public int minimumTotal(List<List<Integer>> triangle) {

        int[] dp = new int[triangle.size()];

        // 创建 DP 空间

        for (int i = 0; i < triangle.size(); i++) {

            dp[i] = triangle.get(triangle.size() - 1).get(i);

        }

        for (int i = triangle.size() - 2; i >= 0; i--) {

            for (int j = 0; j <= i; j++) {

                dp[j] = Math.min(dp[j], dp[j+1]) + triangle.get(i).get(j);

            }

        }

        return dp[0];

    }

}

深圳网站建设www.sz886.com

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,772评论 6 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,458评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,610评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,640评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,657评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,590评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,962评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,631评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,870评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,611评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,704评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,386评论 4 319
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,969评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,944评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,179评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 44,742评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,440评论 2 342

推荐阅读更多精彩内容