输入时两个整数数组,他们任意两个数的和又可以组成一个数组,求这个和中前k个大的数。
如题,首先想到的暴力排序,把两个数组能相加得所有数放入一个新的数组,然后用各种排序,假设第一个数组长m,第二个数组长n。若用快速排序或堆排序等较快的排序。时间复杂度为o((log2(m+n))*(m+n)),空间复杂度为o(m+n)。显然我们只需要前k个大的数,并不需要存储所有相加得到的和,当k足够小时,空间有一定浪费。所以我的思路是将相加和的前k个数先塞进k大小的数组并通过快速排序将新数组调整为有序数组。然后将后面m*n-k个和依次插入数组,多出的一位舍弃。插入的方法我使用的是二分法。时间复杂度为o((log2(k))*(m+n)),空间复杂度为o(k)。
代码如下:
#include<iostream>
usi