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;    }};