PAT A1048 二分/two points

weixin_33766168發表於2019-02-16

clipboard.png
簡單的一道題,想到的有兩種方法

方法一:二分查詢;
由於是兩個數和,所以我們從i=1開始列舉,在剩下的i+1~n序列中找到m-a[i]的數,是一個遞增不重複序列的查詢問題,之前二分法總結過,所以不再贅述;
程式碼如下:

#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<algorithm>
using namespace std;
const int maxn=100010;
int coin[maxn];
int n,m;

int binarySearch(int l,int r,int x){
    while(l<=r){
        int mid=(l+r)/2;
        if(coin[mid]==x)
            return mid;
        else if(coin[mid]>x){
            r=mid-1;
        }else{
            l=mid+1;
        }
    }
    return -1;
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&coin[i]);
    }
    int a,b;
    int r=-1;
    sort(coin+1,coin+n+1);
    for(int i=1;i<n;i++){
        a=coin[i];
        b=m-coin[i];
        r=binarySearch(i+1,n+1,b);
        if(r!=-1){
            break;
        }
    }
    if(r==-1){
        printf("No Solution\n");
    }else{
        printf("%d %d",a,b);
    }
    system("pause");
    return 0;
}

第二種方法:two points;
注意的點是避免i=j使得重複加同一個值從而導致錯誤答案;

#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<algorithm>
using namespace std;
const int maxn=100010;
int coin[maxn];
int n,m;

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&coin[i]);
    }
    int a,b;
    int r=-1;
    sort(coin+1,coin+n+1);
    int i=1,j=n;
    int sum;
    bool flag=false;
    while(i<j){
        sum=coin[i]+coin[j];
        if(sum==m){
            flag=true;
            break;
        }
        else if(sum>m){
            j--;
        }else{
            i++;
        }
    }
    if(!flag){
        printf("No Solution\n");
    }else{
        printf("%d %d",coin[i],coin[j]);
    }
    system("pause");
    return 0;
}

相關文章