Leetcode Best Time to Buy and Sell Stock with Cooldown

最新推荐文章于 2020-06-07 23:35:49 发布

原创最新推荐文章于 2020-06-07 23:35:49 发布 · 293 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#dp

leetcode 同时被 2 个专栏收录

167 篇文章

订阅专栏

11 篇文章

订阅专栏

本文介绍了一种使用动态规划解决股票交易问题的方法，旨在计算给定价格序列下，遵循特定交易规则时的最大可能利润。该算法通过维护一个状态数组来记录在不同交易日的最佳买入与卖出策略。

题意：给出每个时刻股票的价格，每天只能买入或卖出，买入或卖出后第二天不能交易。

思路：DP，dp[i][j] 表示在第i天买入，第j天卖出的最大利润，转态方程为：dp[i][j] = max(dp[i][j - 1], max(dp[i - 2][k])。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        vector<int> dp(prices.size(), 0);
        int max_pro = 0;
        vector<int> dp_pre(prices.size(), 0);
        
        for(int i = 0; i < prices.size(); ++ i) {
            for(int j = i; j < prices.size(); j ++) {
                int temp_max = 0;
                if(i - 2 >= 0) temp_max = dp_pre[i - 2];
               
                if(prices[j] - prices[i] > 0) dp[j] = prices[j] - prices[i] + temp_max;
                else dp[j] = temp_max;
                if(j - 1 > 0) dp[j] = max(dp[j], dp[j - 1]);
                max_pro = max(max_pro, dp[j]);
                dp_pre[j] = max(dp_pre[j], dp[j]);
                //printf("%d ", dp[i][j]);
            }
            //cout << endl;
        }
        
        return max_pro;
    }
    
};1