1. 程式人生 > >1267 4個數和為0(不放回的取 4 個)

1267 4個數和為0(不放回的取 4 個)

1267 4個數和為0

  1. 1 秒
  2.  
  3. 131,072 KB
  4.  
  5. 20 分
  6.  
  7. 3 級題

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

輸入樣例

5
-1
1
-5
2
4

輸出樣例

Yes

題解:見程式碼

#include<set>
#include<map>
#include<list>
#include<queue>
#include<stack>
#include<math.h>
#include<vector>
#include<bitset>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#define eps (1e-8)
#define MAX 0x3f3f3f3f
#define u_max 1844674407370955161
#define l_max 9223372036854775807
#define i_max 2147483647
#define re register
#define pushup() tree[rt]=tree[rt<<1]+tree[rt<<1|1]
#define nth(k,n) nth_element(a,a+k,a+n);  // 將 第K大的放在k位
#define ko() for(int i=2;i<=n;i++) s=(s+k)%i // 約瑟夫
using namespace std;

inline int read(){
    char c = getchar(); int x = 0, f = 1;
    while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
    while(c >= '0' & c <= '9') x = x * 10 + c - '0', c = getchar();
    return x * f;
}

typedef long long ll;
const double pi = atan(1.)*4.;
const int M=1e3+5;
const int N=1e6+5;
ll a[1005];
int n;
bool fun(){
    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 sum=a[l]+a[r];
                if(sum==ans)
                   return true;
                else if(sum<ans)
                   l++;
                else
                   r--;
            }
        }
    }
    return false;
}
int main(){
    scanf("%d",&n);
    for(int i=0;i<n;i++)
        scanf("%lld",&a[i]);
    sort(a,a+n);
    if(fun())  printf("Yes\n");
    else printf("No\n");
    return 0;

}