N个数中找三个数，使其和的绝对值最小--

原题：

        Write a function to find any subset of 3 integers from a set of N integers that have a sum with the smallest absolute value, and analyze the time and space complexity of your function.

The input to this function is a pointer or reference to an array of integers, along with the value of N.
When N >= 3, the output is a 3-element array of integers, such that their sum has the smallest absolute value.
When N < 3, the output is null.

For example, given the following input:

{-99, -66, 0, 2, 3}, your function should return {0, 2, 3}.

 

题意即：

       给定包含N个数的无序数组S（可能包含负数，0，正数）。求三个数A,B,C，使其和的绝对值最小。
       例如：S={-9，0，1，3，6}，A=-9,B=3,C=6,MIN=0

 

 

==================================================================

 

 

法一、暴力搜索法，三重循环枚举3个数，时间复杂度为O(N^3)，代码如下：

 

 //时间复杂度是O(n^3) //X,Y,Z中依次存放着三个被选择的数在原数组中的下标 void BrutalDeal(int a[],int n) { int X,Y,Z; int MIN = 1000000; for(int i=0;i<n;i++) { for(int j=i+1;j<n;j++) for(int k=j+1;k<n;k++) if(abs(a[i]+a[j]+a[k])<MIN) { X = i; Y = j; Z = k; MIN = abs(a[i]+a[j]+a[k]); } } cout<<endl<<"MIN = a["<<X<<"]+a["<<Y<<"]+a["<<Z<<"] = "<<a[X]<<"+"<<a[Y]<<"+"<<a[Z]<<" = "<<MIN<<endl; }

 

法二、对数组a排序后枚举其中2个数，二分查找另一个数。O(N^2*LOGN)

 

 

法三、对数组a排序后枚举其中1个数cur（a[i]），使用双向指针head、tail从数组两端更新（注意剔除掉cur）。根据a[head]+a[tail]+cur的正负号来更新head或tail。若为负则head++，否则tail--。一旦和为0即退出。同时根据此和来更新MIN。最后MIN即为所求。利用三个变量X,Y,Z记录任一最优解在排序后数组中的下标。该方法最坏的时间复杂度为O(N^2)。先写一个快速排序算法，因此代码如下：

 

//对a[s]到a[t]的元素进行快速排序 //调用格式：QuickSort(a,0,n-1),a为数组，n为数组长度 void QuickSort(int a[],int s,int t) { int i = s,j = t; int tmp; if(s<t) { tmp = a[s]; //tmp暂存起始元素 while(i != j) { while(j>i && a[j]>=tmp) //从右到左找到第一个比tmp小的数 j--; a[i] = a[j]; //将a[j]移至a[i]的位置上 while(i<j && a[i]<=tmp) //从左到右找到第一个比tmp大的数 i++; a[j] = a[i]; //将a[i]移至a[j]的位置上 } a[i] = tmp; //将起始元素tmp移至a[i]的位置 QuickSort(a,s,i-1); //i为分割点下标，a[i]为分区元素 QuickSort(a,i+1,t); } } void SortDeal(int a[],int n) { //先将数组进行排序 QuickSort(a,0,n-1); for(int i=0;i<n;i++) cout<<a[i]<<" "; //开始进行选择 int MIN = 1000000; int cur; int head,tail; int X,Y,Z; for(int i = 0;i<n;i++) { head = 0; tail = n-1; cur = a[i]; while(head<tail) { if(head == i) head++; //剔除掉cur(a[i]) if(tail == i) tail--; //剔除掉cur(a[i]) int now = cur+a[head]+a[tail]; if(now == 0 || abs(now)<MIN) { MIN = abs(now); X = i; Y = head; Z = tail; if(now == 0) break; } else if(now<0) head++; else tail--; } if(MIN == 0) break; } cout<<endl<<"MIN = a["<<X<<"]+a["<<Y<<"]+a["<<Z<<"] = "<<a[X]<<"+"<<a[Y]<<"+"<<a[Z]<<" = "<<MIN<<endl; }

 

 

 

 

 

以上方法的主函数如下：

 

 

#include <iostream> #include <time.h> using namespace std; int main() { int N; cout<<"Input N: "; cin>>N; int *ap = new int[N]; srand((unsigned)time(NULL)); //随机生成数组ap[N]中的元素 for(int i=0;i<N;i++) { ap[i] = rand(); if(rand()%2) ap[i] *= -1; cout<<ap[i]<<" "; } BrutalDeal(ap,N); SortDeal(ap,N); return 0; }