LeetCode:LCP 24. 数字游戏(对顶堆求中位数 Java)

2024-02-02 21:28

本文主要是介绍LeetCode:LCP 24. 数字游戏(对顶堆求中位数 Java),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

目录

LCP 24. 数字游戏

题目描述:

实现代码与解析:

原理思路:


LCP 24. 数字游戏

题目描述:

        小扣在秋日市集入口处发现了一个数字游戏。主办方共有 N 个计数器,计数器编号为 0 ~ N-1。每个计数器上分别显示了一个数字,小扣按计数器编号升序将所显示的数字记于数组 nums。每个计数器上有两个按钮,分别可以实现将显示数字加一或减一。小扣每一次操作可以选择一个计数器,按下加一或减一按钮。

主办方请小扣回答出一个长度为 N 的数组,第 i 个元素(0 <= i < N)表示将 0~i 号计数器 初始 所示数字操作成满足所有条件 nums[a]+1 == nums[a+1],(0 <= a < i) 的最小操作数。回答正确方可进入秋日市集。

由于答案可能很大,请将每个最小操作数对 1,000,000,007 取余。

示例 1:

输入:nums = [3,4,5,1,6,7]

输出:[0,0,0,5,6,7]

解释: i = 0,[3] 无需操作 i = 1,[3,4] 无需操作; i = 2,[3,4,5] 无需操作; i = 3,将 [3,4,5,1] 操作成 [3,4,5,6], 最少 5 次操作; i = 4,将 [3,4,5,1,6] 操作成 [3,4,5,6,7], 最少 6 次操作; i = 5,将 [3,4,5,1,6,7] 操作成 [3,4,5,6,7,8],最少 7 次操作; 返回 [0,0,0,5,6,7]。

示例 2:

输入:nums = [1,2,3,4,5]

输出:[0,0,0,0,0]

解释:对于任意计数器编号 i 都无需操作。

示例 3:

输入:nums = [1,1,1,2,3,4]

输出:[0,1,2,3,3,3]

解释: i = 0,无需操作; i = 1,将 [1,1] 操作成 [1,2] 或 [0,1] 最少 1 次操作; i = 2,将 [1,1,1] 操作成 [1,2,3] 或 [0,1,2],最少 2 次操作; i = 3,将 [1,1,1,2] 操作成 [1,2,3,4] 或 [0,1,2,3],最少 3 次操作; i = 4,将 [1,1,1,2,3] 操作成 [-1,0,1,2,3],最少 3 次操作; i = 5,将 [1,1,1,2,3,4] 操作成 [-1,0,1,2,3,4],最少 3 次操作; 返回 [0,1,2,3,3,3]。

实现代码与解析:

对顶堆求中位数

class Solution {public int[] numsGame(int[] nums) {final int MOD = 1000000007;PriorityQueue<Integer> lq = new PriorityQueue<>((a, b) -> b - a); // 大根堆PriorityQueue<Integer> rq = new PriorityQueue<>(); // 小根堆int n = nums.length;int[] res = new int[n];long lsum = 0;long rsum = 0;for (int i = 0; i < n; i++) {int t = nums[i] - i;if (i % 2 == 0) {lq.offer(t);lsum += t;int j = lq.peek();lq.poll();lsum -= j;rq.offer(j);rsum += j;res[i] = (int) ((rsum - rq.peek() - lsum) % MOD);} else {rq.offer(t);rsum += t;int j = rq.peek();rq.poll();rsum -= j;lq.offer(j);lsum += j;res[i] =  res[i] = (int) ((rsum - lsum) % MOD);}}  return res;}
}

原理思路:

        题目转化:就是所有nums[i] - i 变成相同值的最小和,而这个相同值就是他们的中位数,把这些数放在数轴上就可以很好的观察出来,也可以作为定理记下来。  

        至于为啥是num[i] - i,可以简单证明一下。例如有一个值 a0, a1, a2,假设最后结果为b , b + 1, b + 2,那么和就为|a0 - (b + 0)| + |a1 - (b + 1)| + |a2 - (b + 2)| 等价于sum|(ai - i - b)|,把ai - i 看成一个整体ti,那么就变成了每一个 ti 到某一个值的距离最小和。这个值就是中位数。  

        所以这题的核心就是求中位数,如何求?

就是对顶堆(双优先级队列lq,rq)

- 中位数左的小数用lq大顶堆维护  

- 中位数右的大数用rq小顶堆维护  

- x 为我们正在遍历的数  

- 如果前缀长度是奇数,此时 lq 和 rq 大小相等,我们先把 x 插入 lq,然后弹出 lq 的堆顶,加到 rq 中。这一操作可以保证,无论 x 是大是小,此时 rq 的堆顶就是中位数。  

- 如果前缀长度是偶数,此时 lq 比 rq 少一个元素,我们先把 x 插入 rq,然后弹出 rq 的堆顶,加到 lq 中。

        当然这里是可以优化的,比如奇数时,lq里的最大值都小于x,那么就直接放进rq就好了,就省去了进lq,再出来进rq的步骤,偶数时同理。

        这题只要想到题目转换的含义就很好写了。

这篇关于LeetCode:LCP 24. 数字游戏(对顶堆求中位数 Java)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/672007

相关文章

Spring Boot整合Redis注解实现增删改查功能(Redis注解使用)

《SpringBoot整合Redis注解实现增删改查功能(Redis注解使用)》文章介绍了如何使用SpringBoot整合Redis注解实现增删改查功能,包括配置、实体类、Repository、Se... 目录配置Redis连接定义实体类创建Repository接口增删改查操作示例插入数据查询数据删除数据更

Java Lettuce 客户端入门到生产的实现步骤

《JavaLettuce客户端入门到生产的实现步骤》本文主要介绍了JavaLettuce客户端入门到生产的实现步骤,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要... 目录1 安装依赖MavenGradle2 最小化连接示例3 核心特性速览4 生产环境配置建议5 常见问题

Java使用Swing生成一个最大公约数计算器

《Java使用Swing生成一个最大公约数计算器》这篇文章主要为大家详细介绍了Java使用Swing生成一个最大公约数计算器的相关知识,文中的示例代码讲解详细,感兴趣的小伙伴可以了解一下... 目录第一步:利用欧几里得算法计算最大公约数欧几里得算法的证明情形 1:b=0情形 2:b>0完成相关代码第二步:加

Java 的ArrayList集合底层实现与最佳实践

《Java的ArrayList集合底层实现与最佳实践》本文主要介绍了Java的ArrayList集合类的核心概念、底层实现、关键成员变量、初始化机制、容量演变、扩容机制、性能分析、核心方法源码解析、... 目录1. 核心概念与底层实现1.1 ArrayList 的本质1.1.1 底层数据结构JDK 1.7

Java Map排序如何按照值按照键排序

《JavaMap排序如何按照值按照键排序》该文章主要介绍Java中三种Map(HashMap、LinkedHashMap、TreeMap)的默认排序行为及实现按键排序和按值排序的方法,每种方法结合实... 目录一、先理清 3 种 Map 的默认排序行为二、按「键」排序的实现方式1. 方式 1:用 TreeM

Java中流式并行操作parallelStream的原理和使用方法

《Java中流式并行操作parallelStream的原理和使用方法》本文详细介绍了Java中的并行流(parallelStream)的原理、正确使用方法以及在实际业务中的应用案例,并指出在使用并行流... 目录Java中流式并行操作parallelStream0. 问题的产生1. 什么是parallelS

Java中Redisson 的原理深度解析

《Java中Redisson的原理深度解析》Redisson是一个高性能的Redis客户端,它通过将Redis数据结构映射为Java对象和分布式对象,实现了在Java应用中方便地使用Redis,本文... 目录前言一、核心设计理念二、核心架构与通信层1. 基于 Netty 的异步非阻塞通信2. 编解码器三、

SpringBoot基于注解实现数据库字段回填的完整方案

《SpringBoot基于注解实现数据库字段回填的完整方案》这篇文章主要为大家详细介绍了SpringBoot如何基于注解实现数据库字段回填的相关方法,文中的示例代码讲解详细,感兴趣的小伙伴可以了解... 目录数据库表pom.XMLRelationFieldRelationFieldMapping基础的一些代

一篇文章彻底搞懂macOS如何决定java环境

《一篇文章彻底搞懂macOS如何决定java环境》MacOS作为一个功能强大的操作系统,为开发者提供了丰富的开发工具和框架,下面:本文主要介绍macOS如何决定java环境的相关资料,文中通过代码... 目录方法一:使用 which命令方法二:使用 Java_home工具(Apple 官方推荐)那问题来了,

Java HashMap的底层实现原理深度解析

《JavaHashMap的底层实现原理深度解析》HashMap基于数组+链表+红黑树结构,通过哈希算法和扩容机制优化性能,负载因子与树化阈值平衡效率,是Java开发必备的高效数据结构,本文给大家介绍... 目录一、概述:HashMap的宏观结构二、核心数据结构解析1. 数组(桶数组)2. 链表节点(Node