51nod 1001 数组中和等于K的数对 (二分法)

 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);
    }

}

猜你喜欢

转载自blog.csdn.net/JKdd123456/article/details/82915191
今日推荐