1. 程式人生 > >leetcode -初級 買賣股票的最佳時機 II

leetcode -初級 買賣股票的最佳時機 II

給定一個數組,它的第 i 個元素是一支給定股票第 i 天的價格。

設計一個演算法來計算你所能獲取的最大利潤。你可以儘可能地完成更多的交易(多次買賣一支股票)。

注意:你不能同時參與多筆交易(你必須在再次購買前出售掉之前的股票)。

輸入: [7,1,5,3,6,4]
輸出: 7
解釋: 在第 2 天(股票價格 = 1)的時候買入,在第 3 天(股票價格 = 5)的時候賣出, 這筆交易所能獲得利潤 = 5-1 = 4 。
     隨後,在第 4 天(股票價格 = 3)的時候買入,在第 5 天(股票價格 = 6)的時候賣出, 這筆交易所
輸入: [1,2,3,4,5]
輸出: 4
解釋: 在第 1 天(股票價格 = 1)的時候買入,在第 5 天 (股票價格 = 5)的時候賣出, 這筆交易所能獲得利潤 = 5-1 = 4 。
     注意你不能在第 1 天和第 2 天接連購買股票,之後再將它們賣出。
     因為這樣屬於同時參與了多筆交易,你必須在再次購買前出售掉之前的股票。
輸入: [7,6,4,3,1]
輸出: 0
解釋: 在這種情況下, 沒有交易完成, 所以最大利潤為 0。
class Solution {
public:
    int maxProfit(vector<int>& prices) {
//n 判斷是否購買了股票 以及 儲存購買股票時的價格
//n=-1意味著沒有購買股票

        int n = -1, size = prices.size(), sum = 0;
        if(size == 0)
            return 0;
        prices.push_back(prices[size-1]);
        for(int i = 0; i < size; i++){
            if( prices[i] < prices[i+1] ){
                 if( n == -1)
                    n = prices[i];
            }
            else{
                if(n != -1)
                {
                    sum += prices[i]-n;
                    n = -1;
                }
            }
        }
        return sum;
    }
};

反思:思路為判斷當前價格prices[i]和明天的價格prices[i+1]

             prices[i]<prices[i+1] ?購買:不購買

         問題:遍歷到邊界時的處理,即 i=prices.size()時 prices[i+1]操作結果不可知

                    起初忽視價格可能為0,將n=0 標誌為 未購票

          解決:將prices[size-1]元素插入到prices的下標為size的位置