两个等长升序序列的中位数

给出两个等长的升序序列,如果两个序列按顺序合并后,他的中位数是中间位置的元素,即第(L + 1) / 2个元素。

O(log2n)的算法。

代码:

#include <stdio.h>
#define MAX 100
int m_search(int *a,int *b,int n) {//总2n个元素 偶数个
    int l1 = 0,r1 = n -1,l2 = 0,r2 = n - 1;
    while(l1 != r1 || l2 != r2) {
        int m1 = (l1 + r1) / 2;
        int m2 = (l2 + r2) / 2;
        if(a[m1] == b[m1]) return a[m1];
        if(a[m1] < b[m2]) {
            if((r1 - l1 + 1) % 2) {//总的奇数个数,保留中间点中间点可能会是中位数
                l1 = m1;//a中部前面的元素比超过一半的元素小 舍去
                r2 = m2;//b中部后面的元素比超过一半的元素大 舍去
            }
            else {//总偶数个数,中间点在前半部分
                r2 = m2;//b后半部分元素比总的一半的元素大,而中位数肯定会在前半部分
                l1 = m1 + 1;//包括m1在内的前半部分比总的一半零一个元素小 不会出现中位数
            }
        }
        else {//相反
            if((r2 - l2 + 1) % 2) {
                l1 = m1;
                l2 = m2;
            }
            else {
                r1 = m1;
                l2 = m2 + 1;
            }
        }
    }
    return a[l1] < b[l2] ? a[l1] : b[l2];
}
int main() {
    int n,a[MAX],b[MAX];
    scanf("%d",&n);
    for(int i = 0;i < n;i ++) {
        scanf("%d",&a[i]);
    }
    for(int i = 0;i < n;i ++) {
        scanf("%d",&b[i]);
    }
    printf("%d",m_search(a,b,n));
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/8023spz/p/11838668.html