1. 程式人生 > >51nod 1001 陣列中和等於K的數對 (二分法)

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

}