现在的位置: 首页 > 综合 > 正文

找出数组中和为N+1的的所有组合

2013年02月01日 ⁄ 综合 ⁄ 共 1166字 ⁄ 字号 评论关闭

一个整数数列,元素取值可能是1~N(N是一个较大的正整数)中的任意一个数,相同数值不会重复出现。
设计一个算法,找出数列中符合条件的数对的个数,满足数对中两数的和等于N+1。

复杂度最好是O(n),如果是O(n2)则不得分


方法一:排序后,定义两个指针begin,end分别指向数组的第一个和最后一个元素。然后按以下规则移动指针:

1.  如果*begin + *end = N+1,那么找到一个解,同时begin++,end--

2.  如果*begin + *end > N+1,那么说明不等式后项应该减小,所以end--;

3.  如果*begin + *end < N+1,那么说明不等式前项应该增大,所以begin++;

4.  重复以下过程直到end<begin

 

代码如下:

int main()
{
	int a[] = {1,3,4,7,8,9,10};
	int sum = 11;
	int len = sizeof(a)/sizeof(int);
	sort(a,a+len);
	int high = len-1;
	int low = 0;
	while(low<high)
	{
		if(a[low]+a[high] == sum)
		{
			cout<<a[low]<<" + "<<a[high]<<" = "<<sum<<endl;
			low++;
			high--;
		}else if(a[low]+a[high]<sum)
		{
			low++;
		}else{
			high--;
		}
	}
	getchar();
	return 0;
}

方法二:构建一个辅助数组B,b[i] = N+1 – a[i].然后建立两个指针分别指向数组a的头部(begin),和数组b的尾部(end)。按以下规则移动指针:

1.  如果a[begin] == b[end],那么找到一组解a[end]+b[end]=N+1,begin++,end--;

2.  如果a[begin] > b[end],那么end--;

3.  如果a[begin] < b[end],那么begin++;

4.  直到begin>end;

时间复杂度为O(N),空间复杂度O(N);

using namespace std;
int main()
{
	int a[] = {1,3,4,7,8,9,10};
	int sum = 11;
	int len = sizeof(a)/sizeof(int);
	int *b = (int *)malloc(sizeof(len));
	for(int i=0;i<len;i++)
		b[i]=sum-a[i];
	int high = len-1;
	int low = 0;
	while(low<high)
	{
		if(a[low]==b[high])
		{
			cout<<a[high]<<" + "<<b[high]<<" = "<<sum<<endl;
			low++;
			high--;
		}else if(a[low] > b[high])
		{
			high--;
		}else{
			low++;
		}
	}
	getchar();
	return 0;
}

抱歉!评论已关闭.