买卖股票要是能这么容易就好了

原文

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

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

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

示例 1

输入:[7,1,5,3,6,4]
输出:5
解释:在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
     注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格;同时,你不能在买入前卖出股票。

这个题的暴力解法是对数组中的每个元素,计算它之后每个元素与它的差,跟踪最大值,但是这样的复杂度是O(n*n)

应该还有办法通过一次循环,找到从in的最大值,然后再循环一次,计算每个位置与最大值的差,这是借鉴之前做过的题的思路,这样复杂度能降低到O(n)

JavaScript实现

/**
 * @param {number[]} prices
 * @return {number}
 */
var maxProfit = function(prices) {
    const max = Array(prices.length).fill(0);
    let m = 0;
    for (let i = prices.length - 1; i > -1; i--) {
        m = Math.max(m, prices[i]);
        max[i] = m;
    }

    let best = 0;
    for (let i = 0; i < prices.length; i++) {
        best = Math.max(max[i] - prices[i], best);
    }
    return best;
};

上一篇 下一篇