51nod1001陣列中和等於K的數對
阿新 • • 發佈:2018-12-01
題目描述
給出一個整數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 8Output示例
-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");
}