51NOD1267(4个数和为0)

最新推荐文章于 2021-09-14 20:00:00 发布

做不完的ACM

最新推荐文章于 2021-09-14 20:00:00 发布

阅读量198

点赞数

CC 4.0 BY-SA版权

分类专栏：二分+双指针

本文链接：https://blog.youkuaiyun.com/qq_36876305/article/details/79810679

二分+双指针专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一个经典的算法问题——从N个整数中找出四个数，使其和为零。通过两两组合并利用双指针技术进行查找，文章详细展示了如何高效地解决这一问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给出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示例

Output示例

Yes

两两加入数组，然后双指针往中间走。

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
typedef long double ld;

#define x0 x0___
#define y0 y0___
#define pb push_back
#define SZ(X) ((int)X.size())
#define mp make_pair
#define fi first
#define se second
#define pii pair<int,int>
#define pll pair<ll,ll>
#define pli pair<ll,int>
#define pil pair<int,ll>
#define ALL(X) X.begin(),X.end()
#define RALL(X) X.rbegin(),X.rend()
#define rep(i,j,k) for(int i = j;i <= k;i ++)
#define per(i,j,k) for(int i = j;i >= k;i --)
#define mem(a,p) memset(a,p,sizeof(a))


const ll MOD = 1E9 + 7;
ll qmod(ll a,ll b,ll c) {ll res=1;a%=c; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%c;a=a*a%c;}return res;}
ll gcd(ll a,ll b) { return b?gcd(b,a%b):a;}

template<typename T, typename S>
void upmax(T& a,S b){if(a<b) a=b;}
template<typename T, typename S>
void upmin(T& a,S b){if(a>b) a=b;}
template<typename T>
void W(T b){cout << b << endl;}
void gettle() {while(1);}
void getre() {int t=0;t/=t;}


/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

struct node
{
    int p1,p2,key;
    bool operator < (const node & x) const {
        return key < x.key;
    }
    int operator + (const node &x) const {
        return key + x.key;
    }
};
int main()
{
    int n;
    scanf("%d", &n);
    vector<int>a(n);
    rep(i,0,n-1) scanf("%d",&a[i]);
    vector<node>V;
    rep (i,0,n-1) {
        rep (j,i+1,n-1) {
            V.pb({i,j,a[i]+a[j]});
        }
    }
    sort(ALL(V));
    bool ok(0);
    int l = 0, r = SZ(V) - 1;
    while(l < r) {
        if(V[l] + V[r] > 0) r --;
        else if(V[l] + V[r] < 0) l ++;
        else if(V[l] + V[r] == 0 && V[l].p1 != V[r].p1 && V[l].p1 != V[r].p2 && V[l].p2 != V[r].p1 && V[l].p2 != V[r].p2) {
            ok = 1;
            break;
        } else {
            if(V[r-1].key == V[r].key) r --;
            else if(V[l+1].key == V[l].key) l ++;
            else l ++, r --;
        }
    }
    puts(ok ? "Yes" : "No");
    return 0;
}