1. 程式人生 > >51nod 1267 4個數和為0

51nod 1267 4個數和為0

none mes space read 題解 amp char while else

技術分享圖片

【題解】

  先把數字兩兩組合,問題轉化為求a+b=0. 這個問題顯然可以排序後two pointer做。需要註意選出來的兩個數必須由4個不同位置的數組成,不能重復。

技術分享圖片
 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 #define LL long long
 5 #define rg register
 6 #define N 1000010
 7 using namespace std;
 8 int n,tot,a[N],b[N],pos[N][2];
 9 inline int
read(){ 10 int k=0,f=1; char c=getchar(); 11 while(c<0||c>9)c==-&&(f=-1),c=getchar(); 12 while(0<=c&&c<=9)k=k*10+c-0,c=getchar(); 13 return k*f; 14 } 15 int main(){ 16 n=read(); 17 for(rg int i=1;i<=n;i++) b[i]=read(); 18 for(rg int i=1
;i<n;i++) 19 for(rg int j=i+1;j<=n;j++) 20 a[++tot]=b[i]+b[j],pos[tot][0]=i,pos[tot][1]=j; 21 sort(a+1,a+1+tot); 22 int l=1,r=tot; 23 while(l<r){ 24 while(a[l]+a[r]>0) r--; 25 if(a[l]+a[r]==0){ 26 for(rg int i=r;i>=l;i--)if(a[l]+a[r]==0
){ 27 if(pos[l][0]==pos[r][0]||pos[l][1]==pos[r][0]||pos[l][0]==pos[r][1]||pos[l][0]==pos[r][1]) continue; 28 puts("YES"); 29 return 0; 30 } 31 else break; 32 } 33 l++; 34 } 35 puts("NO"); 36 return 0; 37 }
View Code

51nod 1267 4個數和為0