BM81 买卖股票的最好时机(二)

原创于 2023-06-03 12:05:18 发布 · 93 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #开发语言

数据结构刷题专栏收录该内容

96 篇文章

订阅专栏

该代码实现了一个动态规划解决方案，计算在允许多次买卖股票的情况下，能获得的最大利润。初始化状态dp[i][0]表示第i天结束时未持有股票的最大利润，dp[i][1]表示持有股票的状态。遍历价格数组，更新每一天的状态，最后返回dp[n-1][0]作为结果。

买卖股票的最好时机(二)_牛客题霸_牛客网 (nowcoder.com)

这里允许多次买卖股票，但是需要买之前需要先卖出。

class Solution {
public:
    /**
     * 
     * @param prices int整型vector 
     * @return int整型
     */
    int maxProfit(vector<int>& prices) {
        // write code here
        int n = prices.size();
        vector<vector<int> > dp(n,vector<int>(2,0));
        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];
    }
};