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