[51nod] 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(N^2LogN)

#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;

int a[1100];

bool binary_find(int l, int r, int x)
{
    while (l <= r) {
        int m = (l+r)>>1;
        if (a[m] > x)
            r = m - 1;
        else if (a[m] < x)
            l = m + 1;
        else
            return 1;
    }
    return 0;
}

int main()
{
    //freopen("1.txt", "r", stdin);
    int n;
    scanf("%d", &n);
    for (int i = 0; i < n; i++)
        scanf("%d", &a[i]);
    sort(a, a+n);
    int flag = 0;
    for (int i = 0; i < n; i++) {
        for (int j = i+1; j < n; j++) {
            int x = -(a[i]+a[j]);
        //    printf("%d %d %d\n", a[i], a[j], x);
            if (binary_find(j+1, n-1, x)) {
                printf("%d %d %d\n", a[i], a[j], x);
                flag = 1;
            }
        }
    }
    if (!flag) printf("No Solution\n");

    return 0;
}

更快的二分 同时搜索两个数

#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;

int a[1100];

int main()
{
    //freopen("1.txt", "r", stdin);
    int n;
    scanf("%d", &n);
    for (int i = 0; i < n; i++)
        scanf("%d", &a[i]);
    sort(a, a+n);

    int flag = 0;
    for (int i = 0; i < n; i++) {
        int j, k, x;
        j = i+1;
        k = n-1;
        while (j < k) {
            x = a[i]+a[j]+a[k];
            if (x < 0)
                j++;
            else if (x > 0)
                k--;
            else {
                printf("%d %d %d\n", a[i], a[j], a[k]);
                flag = 1;
                j++; k--;
            }
        }
    }
    if (!flag) printf("No Solution\n");

    return 0;
}