51nod 1267 4个数和为0_r - 4个数和为0 python-优快云博客

本文链接：https://blog.youkuaiyun.com/Kuguotao/article/details/89402840

该博客探讨了如何判断给定整数数组中是否存在四个数，它们的和等于零。提供了两种策略：一是通过将数组元素两两相加并进行结构体存储和二分查找；二是遍历前两个数并使用二分搜索找到后两个数。

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

收起

输入

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

输出

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

输入样例

输出样例

Yes

思路：1.将数组里的值两两相加存进结构体内，记得也要存下是哪两个数相加得来的。然后二分查找。

#include <cstdio>
#include <queue>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <map>
#include <set>
#include <bitset>
#include <stack>
#define ull unsigned long long
using namespace std;
typedef long long ll;
const ll mod=1000000007;
const int N=1005;
const double pi=acos(-1);
const int inf=0x3f3f3f3f;
int a[2000];

struct Node
{
    int x;
    int y;
    long long  sum;
} s[1000100];
bool cmp(const Node &a,const Node &b)
{
    return a.sum<b.sum;
}
int main()
{
    memset(s,0,sizeof(s));
    int cnt = 0;
    int n;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
    }
    for(int i=0; i<n; i++)
    {
        for(int j=i+1; j<n; j++)
        {
            s[cnt].sum = a[i]+a[j];
            s[cnt].x = i;
            s[cnt].y = j;
            cnt++;
        }
    }
    sort(s,s+cnt,cmp);
    int l=0,r=cnt-1;
    while(l<r)
    {
        ll k = s[l].sum+s[r].sum;
        if(k==0&&s[l].x!=s[r].x&&s[l].x!=s[r].y&&s[l].y!=s[r].y&&s[l].y!=s[r].x)
        {
            printf("Yes\n");
            return 0;
        }
        else if((s[l].sum+s[r].sum)<0)
        {
            l++;
        }
        else
        {
            r--;
        }
    }
    printf("No\n");
    return 0;
}

思路2.遍历前两个，二分扫描后两个。

#include <cstdio>
#include <queue>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <map>
#include <set>
#include <bitset>
#include <stack>
#define ull unsigned long long
using namespace std;
typedef long long ll;
const ll mod=1000000007;
const int N=1005;
const double pi=acos(-1);
const int inf=0x3f3f3f3f;
ll a[N];
bool tao(int n)
{
    for(int i=0;i<n-1;i++)
    {
        for(int j=i+1;j<n;j++)
        {
            ll ans=a[i]+a[j];
            int l=j+1,r=n-1;
            while(l<r)
            {
                ll p=a[l]+a[r];
                if(p+ans==0)
                    return true;
                else if(p+ans<0)
                    l++;
                else r--;
            }
        }
    }
    return false;
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%lld",&a[i]);
    }
    sort(a,a+n);
    if(tao(n))
        printf("Yes\n");
    else
        printf("No\n");
    return 0;
}