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

 

一个整数数列，元素取值可能是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;
}

 

方法三. 用hashmap或者hashset来解决

def combine_sum(nums, target):
    s_set = set()
    for num in nums:
        if target - num in s_set:
            print('%d + %d = %d ' % (num, target - num, target))
        else:
            s_set.add(num)

if __name__ == '__main__':
    nums = [1,3,5,6,7]
    combine_sum(nums, 8)