51nod 1090 3个数和为0（K-SUM问题）

 

题目描述：

                                 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；