L - 4个数和为0 51Nod - 1267

给出N个整数，你来判断一下是否能够选出4个数，他们的和为0，可以则输出"Yes"，否则输出"No"。

Input

第1行，1个数N，N为数组的长度(4 <= N <= 1000) 
第2 - N + 1行：Aii（-10^9 <= Aii <= 10^9)

Output

如果可以选出4个数，使得他们的和为0，则输出"Yes"，否则输出"No"。

Sample Input

5
-1
1
-5
2
4

Sample Output

Yes

在暴力的基础上加一个二分优化就行了

953ms

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<queue>
#include<vector>
#include<stack>

using namespace std;
typedef long long ll;
ll n, k;
ll a[2000];
//int vis[60000];
int flag1;
int tt;
void f(ll x, int l, int r)
{
    int m = (l + r) / 2;
    if(l > r)return;
    if(a[m] == x)
    {
        flag1 = 1;
        return;
    }
        if(a[m] < x)f(x, m + 1, r);
        else if(a[m] > x)f(x, l, m - 1);
}
int main()
{
    ios::sync_with_stdio(false);
    int n;
    while(cin >> n){
        for(int i = 0; i < n; i++)cin>>a[i];
        sort(a, a + n);
        int flag = 0;
        for(int i = 0; i < n; i++){
            for(int j = i + 1; j < n; j++){
                for(int k = j + 1; k < n; k++){
                    flag1 = 0;
                    f(0 - a[i] - a[j] - a[k], k + 1, n - 1);
                    if(flag1){
                        flag = 1;
                        break;
                    }
                }
                if(flag)break;
            }
            if(flag)break;
        }
        if(flag)cout << "Yes" << endl;
        else cout << "No" << endl;
    }
    return 0;
}