2020-02-27 刷题 4(动态规划)

70 爬楼梯

比较简单的动态规划题,爬到第n阶的方法分为两类,要么是从n-1阶跨1步上来,要么是从n-2阶跨2步上来。用O(n)表示爬到n的方法,递推公式:

O(n) = O(n-1) + O(n-2)
O(1) = 1, O(2) = 2

代码:

class Solution {
public:
    int climbStairs(int n) {
        vector<int> O;
        O.push_back(0);
        O.push_back(1);
        O.push_back(2);
        for(int i = 3; i <= n; i++)
            O.push_back(O[i-1] + O[i-2]);
        return O[n];
    }
};

121 买卖股票的最佳时机

这个题之前做过一个可以多次买卖的版本(买卖股票的最佳时机II),那个当时还是想了挺久的。今天这个只能买卖一次的,开始想复杂了,差点写个O(n^2)的出来,后来才想到应该怎么做。这个题目上耽误的时间有点多,还是不熟练啊。

其实很简单,只要能得到当前位置前的最小元素,就可以得到如果在当前位置卖出,能获取的最大利润。所以设置两个全局变量,一个是到当前位置为止的能获取的最大利润,一个是到当前位置位置前的最小元素,只要扫描一遍数组,同步更新这两个值就可以,甚至都不用递推方程和数组来存储结果。

这种做法的时间复杂度O(n),空间复杂度O(1),而且代码看起来也很简洁,我觉得还行。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(prices.size() <= 1) return 0;
        int max_profit = 0, cur_min = prices[0];
        for(int i = 1; i < prices.size(); i++){
            if(prices[i] - cur_min > max_profit)
                max_profit = prices[i] - cur_min;
            if(prices[i] < cur_min)
                cur_min = prices[i];
        }
        return max_profit;
    }
};

53 最大子序和

还是典型的动态规划题目,大二上算法课的时候第二个例题就是它。
用dp(i)表示以第i个元素结尾且包含第i个元素的最大子序和,递推公式:

dp(i) = max{num[i], dp(i-1) + num[i]}, i > 0
dp(0) = num[0]

最后扫描一遍dp数组,找出最大值就可以。还是O(n)的时间复杂度。
代码:

class Solution {
public:
    int maxSubArray(vector<int>& nums) {
        vector<int> O;
        if(nums.size() == 0) return 0;
        O.push_back(nums[0]);
        for(int i = 1; i < nums.size(); i++){
            int tmp = nums[i] > O[i-1] + nums[i] ? nums[i] : O[i-1] + nums[i];
            O.push_back(tmp);
        }
        int max = O[0];
        for(int i = 1; i < nums.size(); i++){
            if(O[i] > max) max = O[i];
        }
        return max;
    }
};

后来想了一下,这个题目是满足贪心的性质的,对于这种当前步计算结果只依赖于前一步计算结果的最优化问题,其实dp数组也可以不要,设置一个变量保存前一步结果,一个变量保存当前最优结果就可以了。这样只需要扫描一遍数组,并且空间复杂度是O(1),代码也变得更加简洁了,非常棒棒。

class Solution {
public:
    int maxSubArray(vector<int>& nums) {
        int cur_max = nums[0], last_max = nums[0];
        if(nums.size() == 0) return 0;
        for(int i = 1; i < nums.size(); i++){
            last_max = nums[i] > last_max + nums[i] ? nums[i] : last_max + nums[i];
            cur_max = last_max > cur_max ? last_max : cur_max;
        }
        return cur_max;
    }
};

题目还说能不能用更精妙的分治法来解,我觉得一点都不精妙好不好,解起来很麻烦,而且时间复杂度也没有变好,这种问题复杂度下界就是O(n)了,所以我就不尝试了,以后有时间再说吧。


198 打家劫舍

开始以为是个背包问题,后来发现没有背包,就是个普通的贪心问题,只不过当前的最优结果取决于前两步的最优结果,要多保存个变量。
当前的抢劫结果,就是抢或者不抢两种情况,抢的话,就是dp[i-2]+num[i], 不抢的话,就是dp[i-1]。两个里面取最大。

递推公式:

dp[i] = max{dp[i-1], dp[i-2]+num[i]}

代码:

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