LintCode_149_买卖股票的最佳时机

问题描述

假设有一个数组，它的第i个元素是一支给定的股票在第i天的价格。如果你最多只允许完成一次交易(例如,一次买卖股票),设计一个算法来找出最大利润。

样例

给出一个数组样例 [3,2,3,1,2], 返回 1

求解

此问题是“最大子数组”问题的一个应用。

class Solution {
public:
    /*
    * @param prices: Given an integer array
    * @return: Maximum profit
    */
    int maxProfit(vector<int> prices) {
        // write your code here
        if (prices.size() < 2) return 0;
        int max = 0;
        int sum = 0;
        auto begin = prices.cbegin() + 1;
        auto end = prices.cend();
        for (auto it = begin; it != end; ++it)
        {
            int diff = *it - *(it - 1);
            //if (it == begin) max = diff;    //如果考虑增量全为负数的情况，这一行不加注释
            sum += diff;
            if (max < sum) max = sum;
            if (sum < 0) sum = 0;
        }
        return max;
    }
};