[51nod] 1267 4个数和为0 暴力+二分

给出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"。

Input示例

5
-1
1
-5
2
4

Output示例

Yes
 
暴力+排序+二分

#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;
 
int a[];
 
int main()
{
    //freopen("1.txt", "r", stdin);
    int n;
    scanf("%d", &n);
    for (int i = ; i < n; i++)
        scanf("%d", &a[i]);
    sort(a, a+n);
 
    for (int i = ; i < n; i++)
        for (int j = i+; j < n; j++) {
            int k, m, sum;
            k = j+;
            m = n-;
            while (k < m) {
                sum = a[i]+a[j]+a[k]+a[m];
                if (sum < ) {
                    k++;
                }
                else if (sum > ) {
                    m--;
                }
                else {
                    k++; m--;
                    printf("Yes\n");
                    return ;
                }
            }
        }
    printf("No\n");
 
    return ;
}

还有个很好的思路，先把数组中任意两个数存起来，再从小到大排序，类似二分的思想从两头开始查找，是否满足sum1+sum2==0 并且两个坐标不重叠，每个点只选一次

#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;
 
int a[];
struct node
{
    int x, y;
    int sum;
}P[*];
 
int cmp(node a, node b)
{
    return a.sum < b.sum;
}
 
int main()
{
    freopen("1.txt", "r", stdin);
    int n;
    scanf("%d", &n);
    for (int i = ; i < n; i++)
        scanf("%d", &a[i]);
 
    int k = ;
    for (int i = ; i < n; i++)
        for (int j = i+; j < n; j++) {
            P[k].x = a[i];
            P[k].y = a[j];
            P[k++].sum = a[i]+a[j];
        }
    sort(P, P+k, cmp);
    int l = , r = k-;
    int flag = ;
    while (l < r) {
        if (P[l].sum+P[r].sum== && P[l].x!=P[r].x && P[l].y!=P[r].y
            && P[l].x!=P[r].y && P[l].y != P[r].x) {
            flag = ;
            break;
        }
        else if (P[l].sum+P[r].sum < )
            l++;
        else r--;
    }
    if (flag) printf("Yes\n");
    else printf("No\n");                
 
    return ;
}