51nod 1001 数组中和等于k的数对

xiaoxiao2021-02-28  17

给出一个整数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 ii <= 10^9)Output第1 - M行:每行2个数,要求较小的数在前面,并且这M个数对按照较小的数升序排列。  如果不存在任何一组解则输出:No Solution。Sample Input 8 9 -1 6 5 3 4 2 9 0 8Sample Output -1 9 0 8 2 6 3 5

从小到大排序,然后依次往后二分搜索

#include<bits/stdc++.h> #include<iostream> #include<algorithm> using namespace std; typedef long long ll; const ll maxn=5e5+50; ll aim,n,cnt,num[maxn]; struct node{ ll a,b; }N[maxn]; int main() { scanf("%lld%lld",&aim,&n); for(ll i=1;i<=n;i++) scanf("%lld",&num[i]); sort(num+1,num+n+1); for(ll i=1;i<=n;i++) { ll l=i,r=n,mid; while(r>=l) { mid=(l+r)/2; if(num[mid]+num[i]>=aim) r=mid-1; else if(num[mid]+num[i]<aim) l=mid+1; //printf("l:%lld r:%lld mid:%lld %lld %lld %lld %lld\n",l,r,mid,num[i],num[l],num[mid],num[r]); } if(num[i]+num[l]==aim && l!=i) { N[++cnt].a=num[i]; N[cnt].b=num[l]; } //printf("sdjhiusuishf\n"); } if(cnt==0) { printf("No Solution\n"); return 0; } for(int i=1;i<=cnt;i++) { printf("%lld %lld\n",N[i].a,N[i].b); } return 0; }
转载请注明原文地址: https://www.6miu.com/read-2450105.html

最新回复(0)