1. 程式人生 > >51nod1001陣列中和等於K的數對

51nod1001陣列中和等於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

真的是大意。。。還以為一級演算法題就是暴力出奇跡。。。。想都沒想就打了個暴力-------結果瘋狂T

這題需要注意的點:

常規暴力做法   很容易想到   兩個for迴圈  每一個a[ i ]  遍歷每一個a[ j ]看看和是否是k

然後因為 陣列中各個數字各不相同  所以 先從小到大排序

然後 i從前往後  j從後往前  (i < j)   向中間逼近尋找

然後第一個優化  j 是從後往前找的  當找到了 a[i]+a[j]==k的時候 就可以break了 因為此時相等  而前邊的都小於此時的a[ j ] ,a[ i ]又不變,所以前邊肯定就沒有答案,直接break。

第二個優化   i 是從前往後找的  比如 陣列 2 3 4 5 6 7 8 9 10   k=10

當我們找到了a[ i ]=3  a[ j ]=7 時  下一次 i  往後移 想要等於k j只能往前移動   那麼  7和7以後的數字都不會存在正解,所以不用浪費時間每次都從最後往前找,每次都從上次的答案往前找就可以了。

 

這其實是剪枝的思想  寫程式碼時候還是要多注意  能優化就優化  

程式碼

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<math.h>
using namespace std;
const int maxn=1e5+7;
const int INF=0x3f3f3f3f;
typedef long long ll;
ll n,k;
ll a[maxn];
int main(){
    scanf("%lld%lld",&k,&n);
    for(int i=0;i<n;i++){
        scanf("%lld",&a[i]);
    }
    sort(a,a+n);
    int f=0,h=n;;
    for(int i=0;i<n;i++){
        for(int j=h-1;j>i;j--){
            if(a[i]+a[j]==k){
                printf("%lld %lld\n",a[i],a[j]);f=1;h=j;break;
            }
        }
    }
    if(f==0)printf("No Solution\n");
}

 

然後看了別人,更優秀的做法 ,線性時間就ok。程式碼簡單易懂,還是自己想的太少。

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<math.h>
using namespace std;
const int maxn=1e5+7;
const int INF=0x3f3f3f3f;
typedef long long ll;
int n,k;
int a[maxn];
int main(){
    scanf("%d%d",&k,&n);
    for(int i=0;i<n;i++){
        scanf("%d",&a[i]);
    }
    sort(a,a+n);
    bool flag=false;
    int i=0,j=n-1;
    while(i<j){
        n=a[i]+a[j];
        if(n==k){
            printf("%d %d\n",a[i],a[j]);i++;j--;flag=true;//等於n 向中間逼近
        }
        else if(n<k) i++;//小於n的話  i向後走
        else j--;//大於n的話  j向前走
    }
    if(!flag)printf("No Solution\n");
}