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;
}