买卖股票的最佳时机 II
给定一个数组,它的第 \(i\) 个元素是一支给定股票第 \(i\) 天的价格。设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
输入: [7,1,5,3,6,4]
输出: 7
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 3 天(股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
随后,在第 4 天(股票价格 = 3)的时候买入,在第 5 天(股票价格 = 6)的时候卖出, 这笔交易所能获得利润 = 6-3 = 3 。
针对本题简单的来说就是贪心的技法对其进行处理,但是最初写的时候对贪心的思路并不清楚,于是按照自己大脑模拟的思路进行了相应的编码,利用双指针的方法进行模拟,得到了最后的结果。
双指针
注意两个点
- 双指针在得到 \(buy_day\) 以及 \(sell_day\) 的边界条件需要注意一下
- 设置 \(pre_buy_day\) 以及 \(pre_sell_day\) 这两个边界条件去重
代码如下
class Solution {
public:
int maxProfit(vector<int>& prices) {
int buy_day = 0; int sell_day = 0;
int days = prices.size();
if (days == 0 || days == 1)
return 0;
int profit = 0;
int pre_buy_day = -1;
int pre_sell_day = -1;
for(int i = 0; i < days; i++){
if ( (i == 0 && prices[i] < prices[i+1]) || ((i > 0 && i + 1 < days) && (prices[i] <= prices[i-1] && prices[i] < prices[i+1])) ){
buy_day = i;
}
if ( (i == days-1 && prices[i] > prices[i-1]) || ((i > 0 && i + 1 < days) && (prices[i] > prices[i-1] && prices[i] >= prices[i+1])) ){
sell_day = i;
}
if (sell_day > buy_day && (pre_buy_day != sell_day && pre_sell_day != sell_day)){
profit += (prices[sell_day] - prices[buy_day]);
}
pre_buy_day = buy_day;
pre_sell_day = sell_day;
}
return profit;
}
};
贪心思想
为了贪心的想法实现,我们需要设置一个贪心的数组。
代码如下:
class Solution {
public:
int maxProfit(vector<int>& prices) {
int result = 0;
for (int i = 1; i < prices.size(); i++) {
result += max(prices[i] - prices[i - 1], 0);
}
return result;
}
};