【剑指offer】和为定值的两个数

转载请注明出处：http://blog.csdn.net/ns_code/article/details/24933341

题目描写叙述：

输入一个递增排序的数组和一个数字S，在数组中查找两个数，是的他们的和正好是S。假设有多对数字的和等于S，输出两个数的乘积最小的。

输入：

每一个測试案例包括两行：

第一行包括一个整数n和k，n表示数组中的元素个数，k表示两数之和。当中1 <= n <= 10^6，k为int

第二行包括n个整数。每一个数组均为int类型。

输出：

相应每一个測试案例，输出两个数，小的先输出。假设找不到，则输出“-1 -1”

例子输入：

6 15
1 2 4 7 11 15

例子输出：

4 11

    思路：

最直接的做法是暴力法，两个for循环，时间复杂度为O（n*n）。可是这样没有充分利用升序数组这一前提。我们假设数组为A，长度为len，给定的和为sum，最好的方法是先用数组的第一个数A[low]和最后一个数A[high]相加，看是否等于sum，假设等于sum。则找到了一组数。返回true，假设大于sum，则将较大的数向前移动一位，即high--，此时变成了第一个和倒数第二个数相加，假设小于sum，则将较小的数向后移动一位，即low++。此时变成了第二个和最后一个数相加。依此类推，假设low==high时还未找到和为sum的一组数，则返回false。

该算法的时间复杂度为O（n），空间复杂度为O（1）。

针对该方法须要给出一些证明，证明例如以下：

对于一个升序数列A1,A2,...Ak k>=3，假设A1+Ak大于sum。那么考察k-1个数对和A1+Ak,A2+Ak,...Ak-1+Ak有sum<A1+Ak<=A2+Ak<=,...<=Ak-1+Ak。也就是说，Ak与数列中其他不论什么数的和都不可能等于sum，因此抛弃Ak这个数，对结果没有影响。A1+Ak假设小于sum的话，同理抛弃A1这个数对结果没有影响。

该方法对随意的整数数组都适合。另外，要输出乘积最小的一组，不是必需将所有的结果保存起来，我们由以下我们高中时很熟悉的数学公式能够证明最左边和最右边的两个符合要求的数的乘积最小。

当a+b = c时，ab<=（a+b）的平方/4，当且仅当a==b时，ab取得最大值，二者相差越远，乘积越小。

    aC代码例如以下：

/****************************************************************************************************
题目：输入一个已经按升序排序过的数组和一个数字，在数组中查找两个数，使得它们的和正好是输入的那个数字。
 
要求时间复杂度是O(n)。假设有多对数字的和等于输入的数字。输出随意一对就可以。
比如输入数组1、2、4、7、11、15和数字15。
 
因为4+11=15，因此输出4和11。
*****************************************************************************************************/
#include<stdio.h>
#include<stdbool.h>
/*
在升序数组A中找出和为sum的随意两个元素。保存在a和b中
*/
bool FindTwoNumSum(int *A,int len,int sum,int *a,int *b)
{
	if(A==NULL || len<2)
		return false;
	int low = 0;
	int high = len-1;
	while(low<high)
	{
		if(A[low]+A[high] == sum)
		{
			*a = A[low];
			*b = A[high];
			return true;
		}
		else if(A[low]+A[high] < sum)
			low++;
		else
			high--;
	}
	return false;
}
 
int main()
{
	int n,k;
	static int A[1000000];
	while(scanf("%d %d",&n,&k) != EOF)
	{
		int i;
		for(i=0;i<n;i++)
			scanf("%d",A+i);
 
		int a,b;
		bool can = FindTwoNumSum(A,n,k,&a,&b);
		if(can)
			printf("%d %d\n",a,b);
		else
			printf("-1 -1\n");
	}
 
	return 0;
}

/**************************************************************

    Problem: 1352

    User: mmc_maodun

    Language: C

    Result: Accepted

    Time:1460 ms

    Memory:4820 kb

****************************************************************/   

我们如今来拓展一下，假设数组是乱序的，并且规定数组中的元素所有为非负整数，相同给定一个数sum，在数组中找出随意两个数。使二者的和为sum。

    因为题目中限定了数组中的元素为非负整数，因此我们能够用哈希数组，开辟一个长度为sum的bool数组B[sum]。并所有初始化为false，对数组A进行一次遍历，假设当前的元素A[i]大于sum。则直接跳过。否则，继续作例如以下推断。假设B[A[i]]为false，则将B[sum-A[i]]置为ture。这样当继续向后遍历时，假设有B[A[i]]为true。则有符合条件的两个数，分别为A[i]和sum-A[i]。假设遍历A结束后依旧没有发现有B[A[i]]为true的元素。则说明找不到符合条件的元素。

    该算法的时间复杂度为O（n），但空间复杂度为O（sum）。

或者假设知道非负整数数组A的最大值为MAX，则也能够开辟一个MAX大小的bool数组。思路是一样的。

    完整代码例如以下：

/****************************************************************************************************
题目：输入一个无序的非负整数数组和一个数字，在数组中查找两个数，使得它们的和正好是输入的那个数字。
要求时间复杂度是O(n)。
 
假设有多对数字的和等于输入的数字。输出随意一对就可以。
比如输入数组1、7、4、11、6、15和数字15。因为4+11=15。因此输出4和11。
 
*****************************************************************************************************/
#include<stdio.h>
#include<stdlib.h>
 
/*
在无序数组A中找出和为sum的随意两个元素。保存在a和b中
*/
bool FindTwoNumSum(int *A,int len,int sum,int *a,int *b)
{
	if(A==NULL || len<2)
		return false;
	//各元素均被初始化为false的bool数组
	bool *B = (bool*)calloc(sum,sizeof(bool));
	if(B == NULL)
		exit(EXIT_FAILURE);
	int i;
	for(i=0;i<len;i++)
	{
		if(A[i]>sum)
			continue;
		if(B[A[i]] == false)
			B[sum-A[i]] = true;
		else
		{
			*a = A[i];
			*b = sum-A[i];
			free(B);
			B = NULL;
			return true;
		}
	}
	free(B);
	B = NULL;
	return false;
}
 
int main()
{
	int A[] = {19,3,9,7,12,20,17,18,1,16};
	int len = 10;
	int sum = 24;
	int a,b;
	if(FindTwoNumSum(A,len,sum,&a,&b))
		printf("Find two nums,they are:\n%d and %d\n",a,b);
	else
		printf("Not find\n");
	return 0;
}

測试结果：

watermark/2/text/aHR0cDovL2Jsb2cuY3Nkbi5uZXQvbnNfY29kZQ==/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70/gravity/SouthEast" alt="" />