【51nod】--1001 数组中和等于K的数对(sort&&二分)

来源:互联网 发布:史克威尔艾尼克斯 知乎 编辑:程序博客网 时间:2024/05/16 17:54
1001 数组中和等于K的数对
基准时间限制:1 秒 空间限制:131072 KB 分值: 5 难度:1级算法题
 收藏
 关注
给出一个整数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-165342908
Output示例
-1 90 82 6

3 5

思路:sort排序+二分查找,i从0开始++,j从n-1开始--,始终保持i<j,相加等于k时,i++,j--;相加小于k时,i++;相加大于k时,j--;用flag标记是否有结果输出。

代码如下:

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;int main(){int k,n;int a[50000+11];while(~scanf("%d%d",&k,&n)){int i;for(i=0;i<n;i++)scanf("%d",&a[i]);sort(a,a+n);int j=n-1;i=0;int flag=0;/*for(int i=0;i<n,i<j;i++)//关键在于i和j的范围,不能想当然哦,要考虑清楚了 {if(a[i]+a[j]==k){flag=1;printf("%d %d\n",a[i],a[j]);j--;continue;}else if(a[i]+a[j]>k){while(a[i]+a[j]>k){j--;if(a[i]+a[j]==k){flag=1;printf("%d %d\n",a[i],a[j]);j--;break;}else if(a[i]+a[j]<k)break;}}}(也是正确的方法,和二分思想一样,不过较为麻烦哦)*/while(i<j){if(a[i]+a[j]==k){flag=1;printf("%d %d\n",a[i++],a[j--]);}else if(a[i]+a[j]<k)i++;else j--;}//二分查找哦 if(flag==0)printf("No Solution\n");}return 0;} 

原创粉丝点击