4. 寻找两个有序数组的中位数

最新推荐文章于 2022-10-16 15:42:39 发布

SYaoJun

最新推荐文章于 2022-10-16 15:42:39 发布

阅读量158

点赞数

分类专栏： LeetCode经典题

本文链接：https://blog.csdn.net/SYaoJun/article/details/103616859

版权

LeetCode经典题专栏收录该内容

174 篇文章 3 订阅

订阅专栏

归并排序的扩展

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m = nums1.size();
        int n = nums2.size();
        int i = 0, j = 0;
        int k = (m+n)/2, t = 0, res=0, pre=0;
        while(t <= k && i < m && j < n){
            if(nums1[i] < nums2[j]) {
                pre = res;
                res = nums1[i++];
            }
            else{
                pre = res;
                res = nums2[j++]; 
            }
            t++;   
        }
        while(i == m && t <= k) {
            pre = res;
            res = nums2[j++]; 
            t++;
        }
        while(j == n && t <= k){
            pre = res;
            res = nums1[i++];
            t++;
        }
        
        if((m+n)%2) return res;
        return (res+pre)/2.0;
    }
};