1090 3个数和为0
给出一个长度为N的无序数组,数组中的元素为整数,有正有负包括0,并互不相等。从中找出所有和 = 0的3个数的组合。如果没有这样的组合,输出No Solution。如果有多个,按照3个数中最小的数从小到大排序,如果最小的数相等则按照第二小的数排序。
Input
第1行,1个数N,N为数组的长度(0 <= N <= 1000)
第2 - N + 1行:A[i](-10^9 <= A[i] <= 10^9)
Output
如果没有符合条件的组合,输出No Solution。
如果有多个,按照3个数中最小的数从小到大排序,如果最小的数相等则继续按照第二小的数排序。每行3个数,中间用空格分隔,并且这3个数按照从小到大的顺序排列。
Input示例
7
-3
-2
-1
0
1
2
3
Output示例
-3 0 3
-3 1 2
-2 -1 3
-2 0 2
-1 0 1
题解:
先排序数据,之后o(n2)求前两个数的和,二分去找第三个数存不存在即可。复杂度o(n2logn)
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn = 1000+10;
LL A[maxn];
int n;
bool solve(LL num,int l)
{
int r = n-1;
while(r>=l)
{
int mid=(l+r)>>1;
if(A[mid]==num) return 1;
if(A[mid]<num) l=mid+1;
else r=mid-1;
}
return 0;
}
int main()
{
cin>>n;
for(int i=0;i<n;i++)
cin>>A[i];
sort(A,A+n);
bool flag=1;
for(int i=0;i<n-1;i++)
for(int j=i+1;j<n;j++)
{
int tmp = -A[i]-A[j];
if(tmp<A[j])
break;
if(solve(tmp,j+1))
{
cout<<A[i]<<" "<<A[j]<<" "<<tmp<<endl;
flag=0;
}
}
if(flag) cout<<"No Solution"<<endl;
return 0;
}