【leetcode】122 买卖股票的最佳时机 II(动态规划)

题目链接:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-ii/

题目描述

给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。

注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。

示例 1:

输入: [7,1,5,3,6,4]
输出: 7
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 3 天(股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
     随后,在第 4 天(股票价格 = 3)的时候买入,在第 5 天(股票价格 = 6)的时候卖出, 这笔交易所能获得利润 = 6-3 = 3 。

示例 2:

输入: [1,2,3,4,5]
输出: 4
解释: 在第 1 天(股票价格 = 1)的时候买入,在第 5 天 (股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
     注意你不能在第 1 天和第 2 天接连购买股票,之后再将它们卖出。
     因为这样属于同时参与了多笔交易,你必须在再次购买前出售掉之前的股票。

示例 3:

输入: [7,6,4,3,1]
输出: 0
解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。

思路

1 峰谷法

假设给定的数组为:[7, 1, 5, 3, 6, 4]
如果我们在图表上绘制给定数组中的数字,我们将会得到:
【leetcode】122 买卖股票的最佳时机 II(动态规划)_第1张图片

如果我们分析图表,那么我们的兴趣点是连续的峰和谷

用数学语言描述为:

T o t a l P r o f i t = ∑ i ( h e i g h t ( p e a k i ) − h e i g h t ( v a l l e y i ) ) Total Profit= \sum_{i}(height(peak_i)-height(valley_i)) TotalProfit=i(height(peaki)height(valleyi))

关键是我们需要考虑到**紧跟谷的每一个峰值以最大化利润。**如果我们试图跳过其中一个峰值来获取更多利润,那么我们最终将失去其中一笔交易中获得的利润,从而导致总利润的降低。

例如,在上述情况下,如果我们跳过 p e a k i peak_i peaki
v a l l e y j valley_j valleyj, 试图通过考虑差异较大的点以获取更多的利润,获得的净利润总是会小与包含它们而获得的静利润,因为 C 总是小于 A+B。

复杂度分析
时间复杂度:O(n)
空间复杂度:O(1)

/*
 * 动态规划一次遍历
 * 时间复杂度 O(n) 空间复杂度O(1)
 */
class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(prices.empty()) return 0;

        int peak = prices[0];   // 峰 极大值
        int valley = prices[0]; // 谷 极小值
        int ret = 0;
        int i = 0;
        // 累加每一对紧邻着的波峰-波谷差值
        while (i < prices.size()-1){
            while (i+1 < prices.size() && prices[i] >= prices[i+1]) ++ i;   // 找波谷
            valley = prices[i];
            while (i+1 < prices.size() && prices[i] <= prices[i+1]) ++ i;   // 找波峰
            peak = prices[i];
            if (peak - valley > 0)
                ret += peak - valley;
        }
        return ret;
    }
};

【leetcode】122 买卖股票的最佳时机 II(动态规划)_第2张图片

2 简单的一次遍历

该解决方案遵循 方法1 的本身使用的逻辑,但有一些轻微的变化。在这种情况下,我们可以简单地继续在斜坡上爬升并持续增加从连续交易中获得的利润,而不是在谷之后寻找每个峰值。最后,我们将有效地使用峰值和谷值,但我们不需要跟踪峰值和谷值对应的成本以及最大利润,但我们可以直接继续增加加数组的连续数字之间的差值,如果第二个数字大于第一个数字,我们获得的总和将是最大利润。这种方法将简化解决方案。 这个例子可以更清楚地展现上述情况:

[1, 7, 2, 3, 6, 7, 6, 7]

与此数组对应的图形是:

【leetcode】122 买卖股票的最佳时机 II(动态规划)_第3张图片

从上图中,我们可以观察到 A+B+C的和等于差值 D 所对应的连续峰和谷的高度之差。

复杂度分析
时间复杂度:O(n)
空间复杂度:O(1)

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(prices.empty()) return 0;

        int ret = 0;
        for (int i = 1; i < prices.size(); ++i) {
            if(prices[i] - prices[i-1] > 0)
                ret += prices[i] - prices[i-1];
        }
        return ret;
    }
};

你可能感兴趣的:(LeetCode)