leetcode004：Median of Two Sorted Arrays

问题描述

There are two sorted arrays A and B of size m and n respectively. Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

思路分析

用到归并排序里面的关于两个有序数组合并的思想。

- 如果两数组总数是奇数，则中位数是最中间值，偶数则为最中间两个数均值；
- 可能会出现某一数组已找完还没有到达中位数，需要继续在另一个数组上向后搜寻。

代码

class Solution {
public:
    double findMedianSortedArrays(int A[], int m, int B[], int n) {
        double ans=0;
        int len = (m + n + 1) / 2;
        int flag = (m + n) % 2;     //用来标记中位数是一个数还是两个数的均值（1代表一个数，0代表两数均值）
        int finish = 0;             //用来表示寻找工作是否结束
        int count = 0;
        int i=0,j=0;
        while (i < m&&j < n)
        {
            if (A[i] < B[j])
            { 
                count++; 
                if (count == len) {
                    ans += A[i]; if (flag) { finish = 1; break; }
                }
                if (count == len + 1){ 
                    ans += A[i]; finish = 1; break;
                }
                i++;
            }
            else { 
                count++; 
                if (count == len) {
                    ans += B[j]; if (flag){ finish = 1; break; }
                }
                if (count == len + 1){
                    ans += B[j]; finish = 1; break; 
                }
                j++;
            }
        }
        while (!finish&&i < m)
        {
            count++;
            if (count == len) { ans += A[i]; if (flag) break; }
            if (count == len + 1){ ans += A[i]; break; }
            i++;
        }
        while (!finish&&j < n)
        {
            count++;
            if (count == len) { ans += B[j]; if (flag) break; }
            if (count == len + 1){ ans += B[j]; break; }
            j++;
        }
        ans = flag ? ans : ans / 2;
        return ans;
    }
};