LeetCode题目122. 买卖股票的最佳时机 II
给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例1
输入: [7,1,5,3,6,4]
输出: 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 天接连购买股票,之后再将它们卖出。
因为这样属于同时参与了多笔交易,你必须在再次购买前出售掉之前的股票。
示例3
输入: [7,6,4,3,1]
输出: 0
解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。
解题思路
此题为LeetCode121. 买卖股票的最佳时机的扩展。
与LeetCode121的区别在于不限于一次交易,而是尽可能的多交易获得最大的利润。定义两个变量buy和sell,buy用于记录买入时的最大利润,sell用于记录卖出时的最大利润。利用贪心算法的思路有状态转移方程:
buy=max{buy, sell-price[i]};
sell=max{sell,buy+price[i]};
注意状态方程和LeetCode121的区别,buy的状态转移方程改变了-price[i]
->sell-price[i]
。因为尽可能的多交易以获得最大的利润,所以buy会将之前sell所获得的利润进行累加。
又由于第一天不可能卖出,要想获得最大利润最后一天处于卖出状态,所以有边界条件:
buy=-price[0];
sell=0;
完整代码
public int maxProfit(int[] prices) {
int sell[]=new int[prices.length];
int buy[]=new int[prices.length];
sell[0]=0;
buy[0]=-prices[0];
for (int i = 1; i <prices.length ; i++) {
buy[i]=Math.max(buy[i-1],sell[i-1]-prices[i]);
sell[i]=Math.max(sell[i-1],buy[i-1]+prices[i]);
}
return sell[prices.length-1];
}
由于buy方程中和其相关的buy状态只有buy[i-1];sell方程中和其相关的sell状态只有sell[i-1],所以可以对代码进行优化,减少空间占用,即
public int maxProfit(int[] prices) {
int sell=0;
int buy=-prices[0];
for (int i = 1; i <prices.length ; i++) {
buy=Math.max(buy,sell-prices[i]);
sell=Math.max(sell,buy+prices[i]);
}
return sell;
}
更多LeetCode题目及答案解析见GitHub: https://github.com/on-the-roads/LeetCode
剑指offer题目及答案解析:https://github.com/on-the-roads/SwordToOffer