题目描述
给定两个有序数组arr1和arr2,再给定一个整数K,返回所有数中第K小的数。
[要求]
如果arr1的长度为N,arr2的长度为M,时间复杂度请达到O(\log(\min{N, M}))O(log(minN,M)),额外空间复杂度O(1)O(1)。
输入描述:
第一行三个整数N, M, K分别表示数组arr1, arr2的大小,以及需要询问的数
接下来一行N个整数,表示arr1内的元素
再接下来一行M个整数,表示arr2内的元素
输出描述:
输出一个整数表示答案
我们以下列数组为例子:
arr1={1,2,3,4,5}
arr2={3,4,5}
k=4
我们希望找到第4小的数3,在两个数组中,有三个数字1、2、3小于等于3,所以我们可以把整个数组分成两部分:第一部分包括arr1[0]~arr1[2],arr2[0],其它元素在第二部分,我们定义这两个部分为left和right,left在两个数组中的边界下标为2和0(arr1[2]和arr2[0])。
可以发现,left当中有k个元素,并且left当中所有的元素小于等于right中的元素。在这种情况下,left当中的最大值就是第k小的元素。所以题目可以转化为:保证left当中有k个元素并且left当中的元素全部小于等于right当中的元素,为了满足上述要求,必须找到left的边界下标。
当我们选定一个arr1下标的时候,为了保证left当中包含k个元素,arr2的下标也就随之确定了。如果不满足left<=right,我们可以根据特定的条件判断,调整下标的位置。
我们可以通过二分查找的方式确定arr1的下标,在上述例子中,首先找到arr1的中点下标2,由于left当中要有4个元素,所以arr2的下标为0,此时left<=right,所以left中最大值就是第k小的元素。
我们在以k=7为例,首先在arr1当中,l=0,r=4,找到arr1的中点下标2,由于left当中要有7个元素,arr1提供了3个元素,arr2必须提供4个元素,没有办法实现,此时说明arr1提供的元素太少了,所以l=2+1=3,继续二分查找le

这篇博客介绍了如何在两个已排序的数组中找到第k小的元素。通过二分查找策略,确定数组的边界并确保左边元素小于等于右边元素。文章以实例说明,解释了不同情况下的边界处理和判断规则,最后给出了完整的解决方案代码。

462

被折叠的 条评论
为什么被折叠?



