51nod 1001 陣列中和等於K的數對 (二分法)
阿新 • • 發佈:2018-12-12
1001 陣列中和等於K的數對
給出一個整數K和一個無序陣列A,A的元素為N個互不相同的整數,找出陣列A中所有和等於K的數對。例如K = 8,陣列A:{-1,6,5,3,4,2,9,0,8},所有和等於8的數對包括(-1,9),(0,8),(2,6),(3,5)。
Input
第1行:用空格隔開的2個數,K N,N為A陣列的長度。(2 <= N <= 50000,-10^9 <= K <= 10^9) 第2 - N + 1行:A陣列的N個元素。(-10^9 <= A[i] <= 10^9)
Output
第1 - M行:每行2個數,要求較小的數在前面,並且這M個數對按照較小的數升序排列。 如果不存在任何一組解則輸出:No Solution。
Input示例
8 9 -1 6 5 3 4 2 9 0 8
Output示例
-1 9 0 8 2 6 3 5
思路:
看到題目上邊的提示才知道用二分做
1、輸出的時候需要對兩個數進行排序,就想到用結構體來存兩個數
2、用二分函式 lower_bound() ;
// 返回第一個能插入的位置,如果該位置的值是 k-a[i] 的話,就存入結構體中
3、用了 unique() ; 去重函式,防止出現重複計算
// !!!! 也可以不用去重函式,因為用 lower_bound() 的時候,返回的是你第一個能插入的位置,該位置在之前已經被標記訪問過了,就不會重新計算了
CODE:
#include <iostream> #include <cstdio> #include <cstring> #include <cmath> #include <map> #include <cstdlib> #include <algorithm> using namespace std; typedef long long LL; #define memset(a,n) memset(a,n,sizeof(a)) struct NODE { LL a; LL b; } node[50000+10]; LL a[50000+10]; LL vis[50000+10]; map<LL,LL>m; bool cmp(NODE p,NODE q) { return p.a<q.a; } int main() { LL k,n; memset(a,0); memset(vis,0); scanf("%lld %lld",&k,&n); for(LL i=0; i<n; i++) { scanf("%lld",&a[i]); m[a[i]]++; } LL y; sort(a,a+n); unique(a,a+n); LL x,pos; LL ans=0; LL flag=0; for(LL i=0; i<n; i++) { if(!vis[i]) { vis[i] = 1; x=k-a[i]; pos=lower_bound(a,a+n,x)-a; // cout<<pos<<'*'<<endl; if(a[pos]==x&&vis[pos]==0) { vis[pos]=1; node[flag].a=a[i]; node[flag].b=a[pos]; flag++; } } } if(flag==0) printf("No Solution\n"); else { sort(node,node+flag,cmp); for (LL i=0; i<flag; i++) printf("%lld %lld\n",node[i].a,node[i].b); } }