PAT (Advanced Level) Practice 1029 Median

輸入不要用cin，用scanf，不然會超時。真的要吐血了，沒想到是這個點......
使用cin輸入數據時最後一個測試點超時，使用scanf直接89ms過，題限是200ms
在數據量較大時，用scanf能明顯減少時間...
不過大部分題超時可能還是因爲處理方法不對

這題也不難，使用歸併的思想就能解決，主要還是超時的坑，另外下面還有一份我寫的使用二分的代碼（之前以爲超時是方法不對，所以就寫了二分），也不復雜，速度也是沒話說的。使用scanf是42ms過最後一個用例，哪怕用cin也能100多ms過。這個速度提升就很大了。

#include <iostream>
#include <algorithm>
#include <stdio.h>
#define  maxn 200001
using namespace std;
int a[maxn],b[maxn],la,lb,t[maxn];
int main()
{
    cin>>la;
    for(int i=0;i<la;i++){
        scanf("%d",&a[i]);
    }
    cin>>lb;
    for(int i=0;i<lb;i++){
        scanf("%d",&b[i]);
    }
    int index=(la+lb+1)/2;
    int pa,pb,number;
    pa=pb=0;
    while(pa<la&&pb<lb&&index!=0){
        if(a[pa]>b[pb]){
            number=b[pb];
            pb++;
        }else{
            number=a[pa];
            pa++;
        }
        index--;
    }
    if(index==0){
        cout<<number;
    }else{
       if(pa<la){
            cout<<a[pa+index-1];
       }else if(pb<lb){
            cout<<b[pb+index-1];
       }
    }
    return 0;
}

下面給出我用二分寫的代碼。

#include <iostream>
#include <algorithm>
#include<stdio.h>
#define  maxn 200002
using namespace std;
int a[maxn],b[maxn],la,lb;
bool hasfind=false;
int findMedian(int index,int low,int high,int a[],int b[],int lb){
    while(low<=high){
        int mid = (low+high)/2;
        if(mid>index){
            high=mid-1;
        }else if(index-mid<=lb&&index-mid+1<=lb){
            if(a[mid]>=b[index-mid]&&a[mid]<=b[index-mid+1]){
                    hasfind=true;
                return a[mid];
            }else if(a[mid]<=b[index-mid]){
                low=mid+1;
            }else if(a[mid]>=b[index-mid+1]){
                high=mid-1;
            }
        }else if(index-mid<=lb){
            if(a[mid]>=b[index-mid]){
                hasfind=true;
                return a[mid];
            }else{
                low=mid+1;
            }
        }else{
            return -1;
        }
    }
    return -1;
}
int main()
{
    cin>>la;
    for(int i=1;i<=la;i++){
        scanf("%d",&a[i]);
    }
    cin>>lb;
    for(int i=1;i<=lb;i++){
        scanf("%d",&b[i]);
    }
    int index=(la+lb+1)/2;
    int median= findMedian(index,1,la,a,b,lb);
    if(hasfind==true){
       cout<<median<<endl;
    }else{
        median = findMedian(index,1,lb,b,a,la);
        cout<<median<<endl;
    }
    return 0;
}