[Leetcode] Best Time to Buy and Sell Stock III

最新推荐文章于 2022-07-03 21:43:22 发布

原创最新推荐文章于 2022-07-03 21:43:22 发布 · 745 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

135 篇文章

订阅专栏

本文介绍了一种使用C++编程实现的股票交易算法，通过动态规划和贪婪算法结合，优化了交易策略以最大化利润。

class Solution {
public:
    int maxProfit(vector<int> &prices) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        
        if (a.size() == 0)
            return 0;
        
        vector<int> v1(a.size());
        vector<int> v2(a.size());
        
        int minIdx = 0;
        v1[0] = 0;
        for (int i = 1; i < a.size(); ++i)
        {
            if (a[i] < a[minIdx])
                minIdx = i;
                
            v1[i] = max(v1[i - 1], a[i] - a[minIdx]);
        }
        
        int maxIdx = a[a.size() - 1];
        v2[a.size() - 1] = 0;
        for (int i = a.size() - 2; i >= 0; --i)
        {
            if (a[i] > a[maxIdx])
                maxIdx = i;
            
            v2[i] = max(v2[i + 1], a[maxIdx] - a[i]);
        }
        
        int res = 0;
        for (int i = 0; i < a.size(); ++i)
        {
            res = max(res, v1[i] + v2[i]);
        }
        
        return res;
}

};