算法之求两个数组各任意一位相加和组成的新数组中的前k位

最新推荐文章于 2022-10-27 17:29:32 发布

原创

最新推荐文章于 2022-10-27 17:29:32 发布 · 1.7k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#C++ #算法

输入时两个整数数组，他们任意两个数的和又可以组成一个数组，求这个和中前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>
using namespace std;

int a[6] = { 5, 1, 2, 3, 9, 0 };
int b[8] = { 7, 6, 4, 5, 9, 1, 8, 2 };

//将数字分为两半前面的比关键字大后面的比关键字小
int Partition(int start, int end, int *arr){
	int temp = *(arr + start);
	while (start < end)
	{
		//下面第二个判断条件一定带上=否则数组中有重复元素则死循环
		wh