【剑指offer系列】63-股票的最大利润(大顶堆 + 滑动窗口)

题目:

给定一个数组 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);
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值