51nod 1090 3个数和为0 (二分_stl)

给出一个长度为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

注意没有数据符合的时候要输出no solution！！。。wa1发

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
using namespace std;
#define ll long long
#define N 1005

ll a[N];

int main()
{
#ifdef DONOTKNOW
    freopen("in.txt","r",stdin);
#endif
    ll n;
    scanf("%I64d",&n);
    for(ll i=0;i<n;i++)
    {
        scanf("%I64d",&a[i]);
    }
    sort(a,a+n);
    bool tag=true;
    for(ll i=0;i<n&&a[i]<0;i++)
    {
        for(ll j=i+1,k=n-1;j<n&&j<k;)
        {
            while(a[i]+a[j]+a[k]<0&&j<=k)
            {
                j++;
            }
            while(a[i]+a[j]+a[k]>0&&k>=j)
            {
                k--;
            }
            if(j>=k) break;
            if(a[i]+a[j]+a[k]==0)
            {
                printf("%I64d %I64d %I64d\n",a[i],a[j],a[k]);
                tag=false;
                j++;
                k--;
            }
        }
    }
    if(tag)
    {
        printf("No Solution\n");
    }
    return 0;
}

别人的代码,围观一下

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
using namespace std;
#define ll long long
#define N 1005

ll a[N];

int main()
{
#ifdef DONOTKNOW
    freopen("in.txt","r",stdin);
#endif
    ll n;
    scanf("%I64d",&n);
    for(ll i=0;i<n;i++)
    {
        scanf("%I64d",&a[i]);
    }
    sort(a,a+n);
    bool tag=true;
    for(ll i=0;i<n&&a[i]<0;i++)
    {
        for(ll j=i+1,k=n-1;j<n&&j<k;)
        {
            while(a[i]+a[j]+a[k]<0&&j<=k)
            {
                j++;
            }
            while(a[i]+a[j]+a[k]>0&&k>=j)
            {
                k--;
            }
            if(j>=k) break;
            if(a[i]+a[j]+a[k]==0)
            {
                printf("%I64d %I64d %I64d\n",a[i],a[j],a[k]);
                tag=false;
                j++;
                k--;
            }
        }
    }
    if(tag)
    {
        printf("No Solution\n");
    }
    return 0;
}