算法Day49 | 121. 买卖股票的最佳时机,122.买卖股票的最佳时机II

121. 买卖股票的最佳时机

题目链接:121. 买卖股票的最佳时机
dp数组dp[i][0]表示第i天持有股票得到的最大金额,dp[i][1]表示第i天不持有股票得到的最大金额

递推公式
dp[i][0] = max(dp[i - 1][0], -prices[i])

  • dp[i - 1][0]表示前一天就持有股票,i天没有买入,也没有卖出
  • -prices[i]表示i天买入的股票(股票只能买一次

dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i])

  • dp[i - 1][1]表示前一天就不持有股票,i天没有买入,也没有卖出
  • dp[i - 1][0] + prices[i]表示i - 1天持有股票,第i天卖出

初始化
dp[0][0] = -prices[0], dp[0][1] = 0

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        vector<vector<int>> dp(prices.size(), {-prices[0], 0});//初始化
        for (int i = 1; i < prices.size(); ++i) {
            dp[i][0] = max(dp[i - 1][0], -prices[i]);
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i]);
        }
        return dp.back()[1];//一定是手里没有股票的钱多
    }
};

dp数组可以只用一维数组。原理就像01背包中的二维数组变成一维数组一样。

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

122.买卖股票的最佳时机II

题目链接:122.买卖股票的最佳时机II

与上一题唯一的区别:
递推公式
dp[i][0] = max(dp[i-1][0], dp[i - 1][1] - prices[i])

  • dp[i - 1][0]表示前一天就持有股票,i天没有买入,也没有卖出
  • dp[i - 1][1] - prices[i]表示i天买入的股票
class Solution {
public:
    int maxProfit(vector<int>& prices) {
        vector<vector<int>> dp(prices.size(), {-prices[0], 0});
        for (int i = 1; i < prices.size(); ++i) {
            dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] - prices[i]);
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i]);
        }
        return dp.back()[1];
    }
};

一维数组

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值