给出一个长度为N的无序数组,数组中的元素为整数,有正有负包括0,并互不相等。从中找出所有和 = 0的3个数的组合。如果没有这样的组合,输出No Solution。如果有多个,按照3个数中最小的数从小到大排序,如果最小的数相等则按照第二小的数排序。
#include<iostream>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
using namespace std;
int find(long long *a,long long s,int x,int y)
{ int l=x;
int r=y-1;
while(l<=r)
{
int m=(l+r)/2;
if(a[m]==s)
{
return 1;
}
else if(a[m]<s)
l=l+1;
else if(a[m]>s)
r=r-1;
}
return 0;
}
main()
{
int n;
cin>>n;
long long *a=(long long *)malloc(sizeof(long long )*n);
for(int i=0;i<n;i++)
{
cin>>a[i];
}
sort(a,a+n);
int label=0;
for(int i=0;i<n;i++)
{
for(int j=i+1;j<n;j++)
{ long long ans=0-(a[i]+a[j]);
if(find(a,ans,j+1,n))
{ label=1;
cout<<a[i]<<" "<<a[j]<<" "<<ans<<endl;
}
}
}
if(!label) cout<<"No Solution"<<endl;
}
可将三个数分解为为前两个数的和与第三个数的二分查找,时间复杂度为O(n^2*log2(n)),提高了算法效率。