Leetcode 122 买卖股票的最佳时机②
问题重述
给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。注意: 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例 1:
输入: [7,1,5,3,6,4] 输出: 7
(5-1)+(6-3)= 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 天接连购买股票,之后再将它们卖出。 因为这样属于同时参与了多笔交易,你必须在再次购买前出售掉之前的股票。
思路
贪心算法,取每一次的局部最佳。
比如数组为1436.那么1<4,1买进,4>3那么在4的时候卖出去,一定是最合适的,又因为3<6所以3的时候买进。
所以我们只需要一次循环,只要数组后面一个数b比前一个数大a,那么利润sum又增加了b-a,即 sum += b-a。如果b比a小那么b点就卖出去,那么sum就不增加。
设置两个变量start,end。如果b比a大,那么end = i + 1,sum += prices[end]-prices[start]. 如果b比a小,那么end不更新即可实现。
具体代码
class Solution {
public:
int maxProfit(vector<int>& prices) {
int start = 0, end = 0,sum = 0;
int n = prices.size();
for (int i = 0; i < n-1; i++) {
start = i;
end = i;
if (prices[i+1] > prices[i]) {
end = i+1;
}
sum += prices[end] - prices[start];
}
return sum;
}
};