题目描述:
1090 3个数和为0
给出一个长度为N的无序数组,数组中的元素为整数,有正有负包括0,并互不相等。从中
找出所有和 = 0的3个数的组合。如果没有这样的组合,输出No Solution。如果有多个,按照
3个数中最小的数从小到大排序,如果最小的数相等则按照第二小的数排序。
输入
第1行,1个数N,N为数组的长度(0 <= N <= 1000) 第2 - N + 1行:A[i](-10^9 <= A[i] <= 10^9)输出
如果没有符合条件的组合,输出No Solution。 如果有多个,按照3个数中最小的数从小到大排序,如果最小的数相等则继续按照第二小的数排序。每行3个数, 中间用空格分隔,并且这3个数按照从小到大的顺序排列。输入样例
7 -3 -2 -1 0 1 2 3输出样例
-3 0 3 -3 1 2 -2 -1 3 -2 0 2 -1 0 1
思路:
3-SUM问题,K-SUM问题时间复杂度O(N^(k-1))。
代码实现:
#include<bits/stdc++.h>
#define LL long long
using namespace std;
const int N=2e5+100;
int a[N];
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int n;
while(cin>>n)
{
for(int i=1;i<=n;++i)cin>>a[i];
sort(a+1,a+1+n);
bool f=0;
for(int i=1;i<=n;++i)
{
int l=i+1,r=n;
while(l<r)
{
if((LL)a[l]+a[r]==-a[i])
{
f=1;
cout<<a[i]<<" "<<a[l]<<" "<<a[r]<<endl;
l++;
r--;
}
else if((LL)a[l]+a[r]<-a[i])l++;
else r--;
}
}
if(f==0)cout<<"No Solution"<<endl;
}
return 0;
}
The end;