代码随想录算法训练营Day49 ||121. 买卖股票的最佳时机 ● 122.买卖股票的最佳时机II

这两个题之前做过了,现在重新用动态规划的思路做,并且dp是二维的。

问题1:121. 买卖股票的最佳时机 - 力扣(LeetCode)

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

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

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

思路:该题用暴力的话通过不了,所以得用动规,声明一个二维向量,dp[i][j],其中j只能为0,1,0表示买入,1表示卖出。代码如下:

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

问题2:122. 买卖股票的最佳时机 II - 力扣(LeetCode)

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

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

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

思路:该题则允许进行多次交易,则dp[j]表示第j天的最大利润,代码如下:

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

猜你喜欢

转载自blog.csdn.net/weixin_56969073/article/details/132554008