51nod 1267 4个数和为0 二分查找

题目：

http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1267

题意：

给出N个整数，你来判断一下是否能够选出4个数，他们的和为0，可以则输出”Yes”，否则输出”No”。
Input
第1行，1个数N，N为数组的长度(4 <= N <= 1000)
第2 - N + 1行：A[i]（-10^9 <= A[i] <= 10^9)
Output
如果可以选出4个数，使得他们的和为0，则输出”Yes”，否则输出”No”。

思路：

预处理出任意两个数字之和，并记录下标，然后枚举这个和，去寻找另外一个和，使两者之和为0，注意找出来的下标不能重复

#include <bits/stdc++.h>

using namespace std;

const int N = 1000 + 10, INF = 0x3f3f3f3f;

struct node
{
    int val, i, j;
    friend bool operator< (node a, node b)
    {
        return a.val < b.val;
    }
}arr[N*N];
int a[N];

int main()
{
    int n;
    scanf("%d", &n);
    for(int i = 0; i < n; i++) scanf("%d", &a[i]);
    int k = 0;
    for(int i = 0; i < n; i++)
        for(int j = i+1; j < n; j++)
        {
            arr[k].i = i, arr[k].j = j, arr[k].val = a[i] + a[j], k++;
        }
    sort(arr, arr + k);

    node p;
    bool flag = false;
    for(int i = 0; i < n; i++)
    {
        for(int j = i+1; j < n; j++)
        {
            p.val = 0 - a[i] - a[j];
            int idx = lower_bound(arr, arr + k, p) - arr;
            if(idx >= k) continue;
            while(idx < k && arr[idx].val == p.val)
            {
                if(i != arr[idx].i && i != arr[idx].j && j != arr[idx].i && j != arr[idx].j)
                {
                    flag = true; break;
                }
                idx++;
            }
            if(flag) break;
        }
        if(flag) break;
    }
    puts(flag ? "Yes" : "No");
    return 0;
}