DAY 52 动态规划-CSDN博客

本文链接：https://blog.csdn.net/weixin_45845722/article/details/131565458

动态规划——股票问题

121. 买卖股票的最佳时机

题目描述

给定一个数组 prices ，它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。

你只能选择某一天买入这只股票，并选择在未来的某一个不同的日子卖出该股票。设计一个算法来计算你所能获取的最大利润。

返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润，返回 0 。

思路

通用： dp[i][k][0]/dp[i][k][1]
第i天，最大操作次数k次，0表示不持有股票，1表示持有股票
本题 k = 1，故省去k这一维

状态： 持有/不持有
用 dp[i][0]表示第i天不持有股票时的最大利润；
dp[i][1]表示第i天持有股票时的最大利润；
则所求为max(dp[n-1][0],dp[n-1][1]) = dp[n-1][0]，显然，不持有股票时的利润高

选择：
1.买入——buy；
2.买入的前提下，卖出——sell；
3.不操作——rest

逻辑：
1.第i天不持有股票：dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] + prices[i]);
2.第i天持有股票：dp[i][1] = max(dp[i - 1][1], - prices[i]);

代码


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

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

题目描述

给你一个整数数组 prices ，其中 prices[i] 表示某支股票第 i 天的价格。

在每一天，你可以决定是否购买和/或出售股票。你在任何时候最多只能持有一股股票。你也可以先购买，然后在同一天出售。

返回你能获得的最大利润。

思路

通用： dp[i][k][0]/dp[i][k][1]
第i天，最大操作次数k次，0表示不持有股票，1表示持有股票
本题每天都可以操作，不限制操作次数，故省去k这一维

选择：
1.买入——buy；
2.买入的前提下，卖出——sell；
3.不操作——rest

逻辑：
1.第i天不持有股票：dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] + prices[i]);
2.第i天持有股票：dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] - prices[i]);

代码


// dp[i][0]第i天不持有股票的最大利润
// dp[i][1]第i天持有股票的最大利润
class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        vector<vector<int>> dp(n, vector<int>(2));
        dp[0][0] = 0;
        dp[0][1] = -prices[0];
        for (int i = 1; i< n; 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[n - 1][0];
    }
};