121. 买卖股票的最佳时机
给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
如果你最多只允许完成一笔交易(即买入和卖出一支股票一次),设计一个算法来计算你所能获取的最大利润。
注意:你不能在买入股票前卖出股票。
示例 1:
输入: [7,1,5,3,6,4]
输出: 5
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格;同时,你不能在买入前卖出股票。
示例 2:
输入: [7,6,4,3,1]
输出: 0
解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。
(来源:力扣(LeetCode))
思路:本题使用动态规划的方法求解。
按照题目要求,股票有买入和卖出两种状态,我们分别设置为buy和sell。
求解状态方程:
(1)对于“买入”状态来说,买之后可以卖出,即进入卖状态;也可以不进行交易,则保持买状态。
(2)对于“卖出”状态来说,卖出股票之后不在进行股票交易,则还保持在卖状态。
(3)如果购买当天的股票,则意味着损失-prices[i];当天卖的话则意味着增加prices[i],当天卖出的总收益为buy + prices[i]。
(4)因此,我们只要考虑当天买和之前买哪个收益更高,当天卖和之前卖哪个收益更高。由此可得状态转移方程:
buy = max(buy, -prices[i])(buy为负数)
sell = max(sell,prices[i] + buy)
(5)初始化buy = -prices[0], sell = 0,最后返回sell即可。
class Solution {
public int maxProfit(int[] prices) {
if (prices.length <= 1) return 0;
int buy = -prices[0], sell = 0;
for (int i = 1; i < prices.length; i++) {
buy = Math.max(buy, -prices[i]);
sell = Math.max(sell, prices[i] + buy);
}
return sell;
}
}