第 k 小的数
一、寻找两个有序数组的中位数
1.1 问题描述
给定两个大小为 m 和 n 的不同时为空的有序数组 nums1 和 nums2。找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
1.2 算法分析
题目要求的时间复杂度是 O(log(m + n)),要产生这样级别的时间复杂度只有采用二分查找法,用分治递归的思路来考虑这个问题。
需要转换题目中求中位数的问题为求第 k 小数的问题。如果 m + n 是奇数,那么寻找第 k = (m + n)/2 + 1 小的数即可;如果长度和是偶数,那么我们还需要寻找第 (m + n)/2 小的数,然后计算两数的平均值。
在求解整个问题的过程中,我们始终需要考虑一个很重要的问题--数组索引越界问题。
下面将详细地分析整个递归流程。
①、首先定义递归函数的作用:寻找两个有序数组 nums1 数组中 [L1, R1] 范围内和 nums2 数组 [L2, R2] 范围内第 k 小的数,k 从 1开始计数。
/**
* L1 nums1数组的寻找范围的左边界
* R1 nums1数组的寻找范围的右边界
* L2 nums2数组的寻找范围的左边界
* R2 nums2数组的寻找范围的右边界
* k 需要寻找第k小的元素
*/
int findKth(int[] nums1, int L1, int R1, int[] nums2, int L2, int R2, int k)
②、用 len1 = R1 - L1 + 1 来记录 nums1 数组中寻找范围的长度,用 len2 = R2 - L2 + 1 来记录 nums2 数组中寻找范围的长度。
③、如果要寻找的 k > len1 + len2,就像只有 3 个数字要找第 4 小的数一样,超出寻找区域,显然无法找到。
④、递归的终止条件:
当 len1 = 0 时,说明只有 nums2 数组中有元素,直接取 nums2[L2 + k - 1] 位元素即可。
当 k = 1 时,说明要取的是两个有序数组中的最小值 MIN(nums1[L1], nums2[L2])。
⑤、递归过程:
由于要求的是第 k 小的数,而且是在两个有序数组中求。划分两个数组时按照 k 值来分。取变量 i = MIN(len1, k/2),之所以这么取,是为了防止 L1 + k/2 - 1 > len1 导致从 nums1 取值越界。再取变量 j = MIN(len2, k/2)。
接下来比较 nums1[L1 + i - 1] 和 nums2[L2 + i - 1] 这两个值。
如果 nums1[L1 + i - 1] <= nums2[L2 + j - 1],显然 nums1 数组中索引为 L1 + i - 1 及之前的元素不可能是中位数,去除 nums1 数组中 [L1, L1 + i - 1] 范围内的元素,缩小了查找范围。我们递归调用该函数,此时在 nums1 中的查找范围变成了 nums1[L1 + i, R1],此时要找的也不应该是第 k 小的元素,因为已经剔除了 i 个比 k 小的元素,因此我们要找的元素变成了第 k - i 小的元素。
如果 nums1[L1 + i - 1] > nums2[L2 + j - 1],同理,nums2 数组中索引为 L2 + j - 1 及之前的元素不可能是中位数,缩小查找范围,剔除了 j 个比 k 小的元素,因此我们要找的元素变成了第 k - j 小的元素。
因为 i + j = MIN(len1, k/2) + MIN(len2, k/2) <= k,所以可以直接判断 [L1, L1 + i - 1] 或者 [L2, L2 + j -1] 区间的元素不可能是中位数。
总结:算法的思想是不断的剔除数据,逐渐逼近第 k 小的数。
1.3 时间复杂度
假设数组长度足够长,每次剔除的元素都是 k/2(i 或者j),显然我们需要 log(k) 次才能找到第 k 小数,这和二分查找法是同理的,而我们要找的 k 值要么是 (m + n)/2 + 1,要么额外再加上 (m + n)/2,因此时间复杂度是 O(log(m + n)) 级别的。
1.4 代码实现
#define MIN(a, b) (a) < (b) ? (a) : (b)
int findKth(int* nums1, int left1, int right1, int* nums2, int left2, int right2, int k)
{
int n1 = right1 - left1 + 1;
int n2 = right2 - left2 + 1;
// 递归退出条件
if(k > n1 + n2) {
return 0; // 实际上 k 不会小于 n1 + n2
}
if(n1 == 0) {
return nums2[left2 + k - 1];
}
else if (n2 == 0) {
return nums1[left1 + k - 1];
}
if(k == 1) {
return MIN(nums1[left1], nums2[left2]);
}
int i = MIN(n1, k / 2);
int j = MIN(n2, k / 2);
// 剔除比第 k 小的数还小的数,逐渐逼近
if(nums1[left1 + i - 1] > nums2[left2 + j - 1]) {
return findKth(nums1, left1, right1, nums2, left2 + j, right2, k - j);
}
else {
return findKth(nums1, left1 + i, right1, nums2, left2, right2, k - i);
}
}
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size)
{
// k = (nums1Size + nums2Size) /2 + 1,因为 k 从 1 开始计数
int mid1 = findKth(nums1, 0, nums1Size - 1, nums2, 0, nums2Size - 1, (nums1Size + nums2Size) / 2 + 1);
// 两个数组总长度是奇数
if((nums1Size + nums2Size) % 2 != 0) {
return mid1;
}
// 两个数组总长度是偶数
else {
// 额外求 (nums1Size + nums2Size) / 2 的值
int mid2 = findKth(nums1, 0, nums1Size - 1, nums2, 0, nums2Size - 1, (nums1Size + nums2Size) / 2);
return (mid1 + mid2) / 2.0;
}
}
二、内容来源
第 k 小的数的更多相关文章
- *HDU2852 树状数组(求第K小的数)
KiKi's K-Number Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)T ...
- 计算序列中第k小的数
作者:jostree 转载请注明出处 http://www.cnblogs.com/jostree/p/4046399.html 使用分治算法,首先选择随机选择轴值pivot,并使的序列中比pivot ...
- [LeetCode] Find K-th Smallest Pair Distance 找第K小的数对儿距离
Given an integer array, return the k-th smallest distance among all the pairs. The distance of a pai ...
- #7 找出数组中第k小的数
「HW面试题」 [题目] 给定一个整数数组,如何快速地求出该数组中第k小的数.假如数组为[4,0,1,0,2,3],那么第三小的元素是1 [题目分析] 这道题涉及整数列表排序问题,直接使用sort方法 ...
- 无序数组求第k大/第k小的数
根据http://www.cnblogs.com/zhjp11/archive/2010/02/26/1674227.html 博客中所总结的7种解法,我挑了其中的解法3和解法6进行了实现. 解法3: ...
- 查找第K小的数 BFPRT算法
出处 http://blog.csdn.net/adong76/article/details/10071297 BFPRT算法是解决从n个数中选择第k大或第k小的数这个经典问题的著名算法,但很多人并 ...
- 基于快速排序思想partition查找第K大的数或者第K小的数。
快速排序 下面是之前实现过的快速排序的代码. function quickSort(a,left,right){ if(left==right)return; let key=partition(a, ...
- 算法---数组总结篇2——找丢失的数,找最大最小,前k大,第k小的数
一.如何找出数组中丢失的数 题目描述:给定一个由n-1个整数组成的未排序的数组序列,其原始都是1到n中的不同的整数,请写出一个寻找数组序列中缺失整数的线性时间算法 方法1:累加求和 时间复杂度是O(N ...
- cogs930找第k小的数(k-th number)
cogs930找第k小的数(k-th number) 原题链接 题解 好题... 终极版是bzoj3065(然而并不会) 先讲这个题... 维护\(n+1\)个值域线段树(用主席树),标号\(0\) ...
- 选择问题(选择数组中第K小的数)
由排序问题可以引申出选择问题,选择问题就是选择并返回数组中第k小的数,如果把数组全部排好序,在返回第k小的数,也能正确返回,但是这无疑做了很多无用功,由上篇博客中提到的快速排序,稍稍修改下就可以以较小 ...
随机推荐
- 利用动态资源分配优化Spark应用资源利用率
背景 在某地市开展项目的时候,发现数据采集,数据探索,预处理,数据统计,训练预测都需要很多资源,现场资源不够用. 目前该项目的资源3台旧的服务器,每台的资源 内存为128G,cores 为24 (co ...
- 简述树,Trie,Avl,红黑树
树的表示方法 在平时工作中通常有2种方式来表示树状结构,分别是孩子链表示法和父节点表示法.光说名词可能无法让人联系到实际场景中,但是写出代码之后大家一定就明白了. 孩子链表示法,即将树中的每个结点的孩 ...
- 04-influxdb基本操作
influxdb基本操作 1. 数据库基本操作 # 创建数据库 > create database db01; # 查看数据库 > show databases; name: databa ...
- 基于 HTML + WebGL 结合 23D 的疫情地图实时大屏 PC 版
前言 2019年12月以来,湖北省武汉市陆续发现了多例肺炎病例,现已证实为一种新型冠状病毒感染引起的急性呼吸道传染病并蔓延全国,肺炎疫情牵动人心,人们每天起来第一件事变成了关注疫情进展,期望这场天灾早 ...
- web自动化原理
在说原理之前我想说下我所理解的selenium: (1).支持多语言,多平台,多浏览器 (2).它是一个工具包 (3).提供所有的网页操作api,是一个功能库 通过selenium来实现web自动化, ...
- activated钩子函数
activated钩子函数是在组件被激活后的钩子函数,mounted是不保证组件在document中,也就是组件还没有被激活,因此可以理解为activated执行在mounted之后. 在跳转传值时, ...
- 菜鸟对java和Go的理解
1.go对比java go通过结构体嵌套+接口实现类似面向对象中的继承和多态.个人认为尤其是go的接口抓住了多态的本质.而Go提倡的面向接口的思想也可能使得架构上更加解耦. 2.关于Go不要通过共享内 ...
- ES6 第七节 ES6中新增的数组知识(1)
目录 ES6 第七节 ES6中新增的数组知识(1) 第七节 ES6中新增的数组知识(1) JSON数组格式转换 Array.of()方法: find()实例方法: ES6 第七节 ES6中新增的数组知 ...
- Deepin中安装使用好用的字典GoldenDict
2020-03-21 23:08:17 不说废话直接来安装步骤: 打开Deepin的应用商店,输入GoldenDict查找: 找到后点击安装,然后等待一小会,电脑提示音告诉你已经安装完成: 然后再 ...
- 小白学 Python 数据分析(19):Matplotlib(四)常用图表(下)
人生苦短,我用 Python 前文传送门: 小白学 Python 数据分析(1):数据分析基础 小白学 Python 数据分析(2):Pandas (一)概述 小白学 Python 数据分析(3):P ...