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