51nod 1001 数组中和等于K的数对二分查找/排序

Posted 2020-10-13 Roni

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
-1
6
5
3
4
2
9
0
8

Output示例

-1 9
0 8
2 6
3 5

【分析】：暴力会tle，二分~可以双指针/STL/二分函数 （http://www.cnblogs.com/wuzetiandaren/p/4259199.html）类似。
【代码】：

#include <bits/stdc++.h>

using namespace std;
const int maxn = 50000+10;
int n,k,L,R;
int a[maxn];
int main()
{
   // freopen("in.txt", "r", stdin);
    while(cin>>k>>n)
    {
        int flag=1;
        for(int i=0;i<n;i++)
        {
            cin>>a[i];
        }
        sort(a,a+n);
        int i=0,j=n-1;
        while(i<j)
        {
            if(a[i]+a[j]==k)
            {
                 L=i;
                 R=j;
                 flag=0;
                printf("%d %d\\n",a[L],a[R]);
            }
            if(a[i]+a[j]>k)
            {
                j--;
            }
            else
            {
                i++;
            }
        }
        if(flag)
            printf("No Solution\\n");
    }
    return 0;
}

双指针

 

#include <bits/stdc++.h>

using namespace std;
const int maxn = 50000+10;
int n,k,t,L,R;
int a[maxn];
int main()
{
   // freopen("in.txt", "r", stdin);
    while(cin>>k>>n)
    {
        int flag=1;
        for(int i=0;i<n;i++)
        {
            cin>>a[i];
        }
        sort(a,a+n);
       for(int i=0;i<n;i++)
       {
           t=lower_bound(a,a+n,k-a[i])-a;
           if(a[t]==k-a[i]&&t>i)
           {
               printf("%d %d\\n",a[i],k-a[i]);
               flag=0;
           }
       }
        if(flag)
            printf("No Solution\\n");
    }
    return 0;
}

stl