[程序员代码面试指南]数组和矩阵问题-找到无序数组中最小的k个数(堆排序)
题目链接
题目描述
从无序序列,找到最小topk个元素。
解题思路
使用大根堆维护最小topk个元素:
- 首先前k个元素建立大根堆(从最后一个非叶节点(数组长度/2-1,结点从0计:大致是最后一个节点j与最后一个非叶节点i满足j=2i+1或j=2i+2,PS数组长度len=j+1,大概是有一些取整的原因设计,总之验证这是对的)至根节点(数组第一个元素)调整)。
- 之后维护这个最小k个元素的大根堆(比较后面的元素与根顶元素,若新元素小则替换掉堆顶元素,并进入调整)。
- 最终堆中元素即为所求。
查找topk时间复杂度:O(nlogk)。
相关知识:堆排序
堆的定义
堆是具有以下性质的完全二叉树:
每个结点的值都大于或等于其左右孩子结点的值,称为大顶堆;或者每个结点的值都小于或等于其左右孩子结点的值,称为小顶堆。
用数组表示一个堆结构,堆的定义就是:(结点从0计)
大根堆:arr[i] >= arr[2i+1] && arr[i] >= arr[2i+2]
小根堆:arr[i] <= arr[2i+1] && arr[i] <= arr[2i+2]
堆排序步骤
第一步:建堆:从最后一个非叶节点(数组长度/2-1,结点从0计。推导:最后一个节点下标j与最后一个非叶节点i的关系满足j=2i+1或j=2i+2=>i=j/2-1或j/2-0.5,)至根节点(数组第一个元素)调整。
第二步:反复执行交换、调整:将堆顶元素与树最后一个叶节点交换,从上至下调整剩余节点为堆(称为筛选);再将堆顶元素与最后一个叶节点交换...直到所有元素组成序列。大根堆对应升序,小根堆对应降序。
堆排序特点
- 时间复杂度:平均、最好、最坏均为O(nlogn)
- 相比快排,堆排序的最坏时间复杂度更优,这是堆排序最大的优点。所以堆排序适合记录数n较大的文件,不适合记录数较小的文件。
- 对深度为K的堆,筛选算法关键字比较次数至多为2(K-1);则在建n个元素,深度为h的堆时,总共进行的关键字比较次数不超过4n(公式见数据结构严蔚敏P282底栏??);又,n个结点的完全二叉树深度为log2n」+1,所以最坏时间复杂度O(nlogn).
堆排序参考链接
https://www.cnblogs.com/chengxiao/p/6129630.html
代码
import java.util.*;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> maxRootHeap = new ArrayList<Integer>();
if(k<1||k>input.length){
return maxRootHeap;
}
for(int i=0;i<k;++i) {
maxRootHeap.add(input[i]);
}
//建堆
buildMaxRootHeap(maxRootHeap);
//调整
for(int i=k;i<input.length;++i) {
if(input[i]<maxRootHeap.get(0)) {
maxRootHeap.set(0, input[i]);
heapify(maxRootHeap,0,k-1);
}
}
return maxRootHeap;
}
private void buildMaxRootHeap(ArrayList<Integer> maxRootHeap) {
for(int i=maxRootHeap.size()/2-1;i>=0;--i) {
heapify(maxRootHeap,i,maxRootHeap.size()-1);
}
}
//调整以index索引为根节点的堆
private void heapify(ArrayList<Integer> maxRootHeap,int index,int heapSize) {//heapSize 指堆最后一个节点的索引
int lIdx=2*index+1;
int rIdx=2*index+2;
int maxIdx=index;
while(lIdx<=heapSize) {
if(maxRootHeap.get(lIdx)>maxRootHeap.get(index)) {
maxIdx=lIdx;
}
if(rIdx<=heapSize&&maxRootHeap.get(rIdx)>maxRootHeap.get(maxIdx)) {
maxIdx=rIdx;
}
if(maxIdx!=index) {
swap(maxRootHeap,index,maxIdx);
}
else {
break;
}
index=maxIdx;
lIdx=2*index+1;
rIdx=2*index+2;
}
}
private void swap(ArrayList<Integer> heap,int idx1,int idx2) {
int temp=heap.get(idx1);
heap.set(idx1,heap.get(idx2));
heap.set(idx2, temp);
}
}
[程序员代码面试指南]数组和矩阵问题-找到无序数组中最小的k个数(堆排序)的更多相关文章
- 《程序员代码面试指南》第八章 数组和矩阵问题 找到无序数组中最小的k 个数
题目 找到无序数组中最小的k 个数 java代码 package com.lizhouwei.chapter8; /** * @Description: 找到无序数组中最小的k 个数 * @Autho ...
- 程序员代码面试指南:IT名企算法与数据结构题目最优解
第1章栈和队列 1设计一个有getMin功能的栈(士★☆☆☆) 1由两个栈组成的队列(尉★★☆☆) 5如何仅用递归函数和栈操作逆序一个栈(尉★★☆☆) 8猫狗队列(士★☆☆☆)10用一个栈实现另一 ...
- 【算法】数组与矩阵问题——找到无序数组中最小的k个数
/** * 找到无序数组中最小的k个数 时间复杂度O(Nlogk) * 过程: * 1.一直维护一个有k个数的大根堆,这个堆代表目前选出来的k个最小的数 * 在堆里的k个元素中堆顶的元素是最小的k个数 ...
- 程序员代码面试指南 IT名企算法与数据结构题目最优解
原文链接 这是一本程序员面试宝典!书中对IT名企代码面试各类题目的最优解进行了总结,并提供了相关代码实现.针对当前程序员面试缺乏权威题目汇总这一痛点,本书选取将近200道真实出现过的经典代码面试题,帮 ...
- 《程序员代码面试指南》第八章 数组和矩阵问题 打印N 个数组整体最大的Top K
题目 打印N 个数组整体最大的Top K java代码 package com.lizhouwei.chapter8; /** * @Description: 打印N 个数组整体最大的Top K * ...
- [程序员代码面试指南]递归和动态规划-最长公共子串问题(DP,LCST)
问题描述 如题. 例:输入两个字符串 str1="1AB234",str2="1234EF" ,应输出最长公共子串"234". 解题思路 状 ...
- [程序员代码面试指南]栈和队列-最大值减去最小值 小于或等于num 的子数组的数量(单调队列)
题目 给定数组arr和整数num,求数组的子数组中有多少个的满足"最大值减去最小值<=num". 解题思路 分析题目,有结论: 如果数组arr[i...j]满足条件,则它的每 ...
- 《程序员代码面试指南》第一章 栈和队列 设计一个有getMin功能的栈
题目 实现一个特殊的栈,在实现栈的基本功能上,再实现返回栈中最小的元素的操作 要求 1. pop.push.getMin操作时间复杂度都是O(1) 2. 设计的栈类型可以使用现成的栈结构 java代码 ...
- 《程序员代码面试指南》第五章 字符串问题 去掉字符串中连续出现k 个0 的子串
题目 去掉字符串中连续出现k 个0 的子串 java代码 package com.lizhouwei.chapter5; /** * @Description: 去掉字符串中连续出现k 个0 的子串 ...
随机推荐
- bitmap--Bitmap详解与Bitmap的内存优化
一.Bitmap: Bitmap是Android系统中的图像处理的最重要类之一.用它可以获取图像文件信息,进行图像剪切.旋转.缩放等操作,并可以指定格式保存图像文件. 常用方法: public voi ...
- Livepeer中文白皮书(翻译)
Livepeer Whitepaper 分布式视频流媒体传输协议及经济激励 Doug Petkanics doug@livepeer.org Eric Tang eric@livepeer.org 翻 ...
- LeetCode【110. 平衡二叉树】
对于平衡二叉树,就是左右深度相差1 就可以另外弄一个函数,计算深度,然后, 在原函数上进行比较深度是否相差1,再输出true or false. 至于迭代就可以,比较完左右节点,再比较各自的左右节点. ...
- C3D视频特征提取
一.部署 1. 先把项目Clone下来 git clone https://github.com/jfzhang95/pytorch-video-recognition.git 2. 安装环境: Py ...
- java 读取外部和source下配置文件
import java.io.File; import java.io.FileInputStream; import java.net.URL; import java.util.Map; impo ...
- C#使用AppDomain时的异常分析:Object ‘XXXX.rem’ has been disconnected or does not exist at the server.
在使用C#的应用程序域的时候,碰到这么一个异常: System.Runtime.Remoting.RemotingException: Object ‘/76e7cd41_2cd2_4e89_9c03 ...
- ring0与ring3通信方式
修改自: https://blog.csdn.net/wzsy/article/details/54929726 控制码方式详解: https://www.cnblogs.com/lsh123/p/7 ...
- LINUX 安装扩展(笔记)
1.下载扩展包: 网址:http://pecl.php.net/ 2.解压安装包. 3.进入解压好的安装包. 4.使用php中的phpize (扩展模块) 使其生成configure. 5../con ...
- gentoo annie youku video
在gentoo 上面,如果需要下载 youku 的视频的话,可以使用 annie 这个软件来下载.annie 软件主页:https://github.com/iawia002/annie#instal ...
- !!常用HTML代码
META: <meta http-equiv="X-UA-Compatible" content="IE=edge,chrome=1"> “IE=e ...