两个有序数组元素之积、和的最小K个值

题目来自阿里面试题目

题目

(1)已知A B两个数组，元素有序，构造新的集合S={x*y | x属于A, y属于B} 求S中最小的k个元素，最优解法。

(2)已知A B两个数组，元素有序，构造新的集合S={x+y | x属于A, y属于B} 求S中最小的k个元素，最优解法。

思路：这两个题本质上属于同一个题目，可以使用同一个思路做。

思路(1) 暴力解决，此时时间复杂度为O(n*m)，空间复杂度O(1).

思路(2) 为最短数组中每一个元素都指定一个游标，之后分别于另外一个数组中元素结合，求解最小值。

时间复杂度:O(k*min(m,n)),k表示取最小的k个数，m和n表示数组长度，min(m,n)表示最短数组。

空间复杂度:O(min(m,n)).

代码：

说明：为了简洁说明思路，代码中没有选择最短数组，元素也没去重。

#include <iostream>
#include <set>
using namespace std;

void FindKSmall(int a[],int m,int b[],int n,int c[],int nTopK)
{
	int count = 0;
	int curMin = 0;
	int* curPos = new int[m];
	memset(curPos,0,sizeof(int) * m);
	while(count < nTopK)
	{
		c[count] = 0x3f3f3f3f;
		for (int i = 0;i < m;i++)
		{
			if (curPos[i] < n && c[count] > a[i] * b[curPos[i]])
			{
				c[count] = a[i] * b[curPos[i]];
				curMin = i;
			}
		}
		curPos[curMin]++;
		count++;
	}
	for (int i = 0;i < nTopK;i++)
	{
		cout<<c[i]<< ",";
	}
	cout<<endl;
	delete []curPos;
	curPos = NULL;
}

void FindKSmall_Force(int a[],int m,int b[],int n,int c[],int nTopK)
{
	
	multiset<int> setTopK;
	for (int i = 0;i < m;i++)
	{
		for (int j = 0;j < n;j++)
		{
			setTopK.insert(a[i] * b[j]);
		}
	}
	for(multiset<int>::iterator it = setTopK.begin(); it != setTopK.end(); ++it) 
	{
		cout<< *it << ",";
	}
	cout<<endl;
}

int main()
{
	const int m = 5;
	int a[m] = {1,2,5,7,9};

	const int n = 4;
	int b[n] = {2,4,6,8};

	const int topK = 5;
	int c[topK];

	FindKSmall(a,m,b,n,c,topK);
	FindKSmall_Force(a,m,b,n,c,topK);

	system("pause");
	return 0;
}

其他方法的详细分析 http://blog.csdn.net/sunnianzhong/article/details/8932374