题目:
给定一个数组 prices ,它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。
你只能选择 某一天 买入这只股票,并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。
返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润,返回 0 。
示例 1:
输入:[7,1,5,3,6,4]
输出:5
解释:在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格;同时,你不能在买入前卖出股票。
示例 2:
输入:prices = [7,6,4,3,1]
输出:0
解释:在这种情况下, 没有交易完成, 所以最大利润为 0。
思路
1、[7,2,5,3,6,0,5],将除了首元素外的其它元素[2,5,3,6,0,5]加入大顶堆[6,5,5,3,2,0]
2、遍历数组
- i = 0,第0天,7买入,最大值6那天卖,收益6 - 7 = -1
- i = 1,第1天,2买入,最大值6那天卖,收益 6 - 2 = 4
- i = 2,第2天,5买入,最大值6那天卖,收益6 - 5 = 1
- i = 3,第3天,3买入,最大值6那天卖,收益 6 - 3 = 3
- i = 4,第4天,6买入,最大值6那天卖(当天买当天卖,收益为0,不影响最终结果),收益6 - 6 = 0
- i = 5,第5天,0买入,此时最大值不再是6了,而是5,收益5 - 0 = 5(收益最大)
public class BestTime {
public static void main(String[] args) {
int[] arr = {7,1,5,3,6,4};
int res = bestDiff(arr);
System.out.println(res);
}
private static int bestDiff(int[] prices) {
if (prices.length == 1) {
return 0;
}
Queue<Integer> queue = new PriorityQueue<>((o1, o2) -> o2 - o1);
for (int i = 1; i < prices.length; i++) {
queue.add(prices[i]);
}
int res = Integer.MIN_VALUE;
for (int i = 0; i < prices.length - 1; i++) {
Integer maxVal = queue.peek();
res = Math.max(res, maxVal - prices[i]);
// 第四天是最大值,后续从第五天开始,最大值就不是第四天的值了,需要弹出来首元素
if (maxVal == prices[i + 1]) {
queue.poll();
}
}
return Math.max(res, 0);
}
}