leetcode122-买卖股票的最佳时机

题目

点击去做题

给定一个数组 prices ，其中 prices[i] 是一支给定股票第 i 天的价格。

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

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

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

题解

分两步：

1. 买股票：确定什么时候买入股票。（注意：手中只能同时持有一个股票）

   在第二天的股价比当天股价高时，选择买入

2. 卖股票：确定什么时候卖股票。（注意：手中有股票时才能卖股票）

   当第二天股票价格比当天低时，选择卖出股票；如果一直上升，则在最后一天卖出

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int len = prices.size();
        bool haveStock = 0;         // 手中是否有股票
        int stock=0, profit=0;       // stock为手中持有的股票，profit：到目前为止获得的总利润
        for(int i=0; i<len-1; ++i){
            if(prices[i]<prices[i+1] && !haveStock){   //买股票
                haveStock = 1;
                stock = prices[i];
            }
            if(haveStock && prices[i]>prices[i+1]){   //卖股票
                haveStock = 0;
                profit += prices[i] - stock;
            }
            else if(haveStock && i==len-2){
                haveStock = 0;
                profit += prices[i+1] - stock;
            }

        }
        return profit;
    }
};