浙大PAT 1029题 1029. Median

#include<stdio.h>
#include<stdlib.h>
#define maxn 1000005
long a[maxn],b[maxn];
int main(){
    long i,j,n,m;
    scanf("%ld",&n);
    for(i=0;i<n;i++){
        scanf("%ld",&a[i]);   
    }
    scanf("%ld",&m);
    for(i=0;i<m;i++){
        scanf("%ld",&b[i]);   
    }
    long med;
    if((n+m)%2==0){
       med=(n+m)/2;            
    }
    else{
       med=(n+m+1)/2;      
    }
    long cnt=0,last;
    i=0;j=0;
    while(cnt<med){
        if(i<n&&j<m){
           if(a[i]<b[j]){
               last=a[i];  
               i++;               
           }  
           else{
               last=b[j];  
               j++;  
           }           
        }
        else if(i<n&&j>=m){
            last=a[i];  
            i++;      
        }           
        else if(i>=n&&j<m){
            last=b[j];  
               j++;   
        }     
        cnt++; 
    } 
    printf("%ld\n",last);
    system("pause");
    return 0;
}