题目描述
给定两个有序数组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的下标