两个有序序列的中位数

两个有序序列的中位数

#include<iostream>
#include<stdio.h>
using namespace std;

int M_search(int A[], int B[], int n){
    int s1=0, d1=n-1, m1;
    int s2=0, d2=n-1, m2;
    while(s1 != d1 || s2 != d2){
        m1 = (s1+d1)/2;
        m2 = (s2+d2)/2;
        if(A[m1] == B[m2])
            return A[m1];
        if(A[m1] < B[m2]){
            if((s1+d1)%2 == 0){ //元素个数为奇数
                s1 = m1;        //舍弃A中间点以前的部分且保留中间点
                d2 = m2;        //舍弃B中间点以后的部分且保留中间点
            }
            else{               //元素个数为偶数
                s1 = m1+1;      //舍弃A中间点以及中间点以前的部分
                d2 = m2;        //舍弃B中间点以后的部分且保留中间点
            }
        }
        else{
            if((s2+d2)%2 == 0){
                s2=m2;
                d1=m1;
            }
            else{
                s2=m2+1;
                d1=m1;
            }
        }
    }
    return A[s1] < B[s2] ? A[s1]:B[s2];
}

int main(){
   int A[5] = {11, 13, 15, 17, 19};
   int B[5] = {2, 4, 6, 8, 20};
   printf("%d\n", M_search(A, B, 5));
}