Median of Two Sorted Arrays（C++寻找两个正序数组的中位数）

最新推荐文章于 2022-04-05 15:30:19 发布

coolsunxu

最新推荐文章于 2022-04-05 15:30:19 发布

阅读量84

点赞数

分类专栏： C++ LeetCode 文章标签： leetcode c++

本文链接：https://blog.csdn.net/coolsunxu/article/details/117522272

版权

C++ 同时被 2 个专栏收录

735 篇文章 18 订阅

订阅专栏

LeetCode

459 篇文章 1 订阅

订阅专栏

（1）排序，记录中间两个数

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int start=0,end=0,count=0;
        int m=nums1.size(),n=nums2.size();
        int mid=(m+n)/2;
        int i=0,j=0,tag=0;
        while(i<m && j<n) {
            if(nums1[i]<=nums2[j]) {
                count++;
                if(count==mid) {
                    start=nums1[i];
                }
                if(count==mid+1) {
                    end=nums1[i];
                    tag=1;
                    break;
                }
                i++;
            } else {
                count++;
                if(count==mid) {
                    start=nums2[j];
                }
                if(count==mid+1) {
                    end=nums2[j];
                    tag=1;
                    break;
                }
                j++;
            }
        }
        
        if(tag==1) {
            if((m+n)%2==0) return ((double)start+end)/2;
            else return (double)end;
        } else {
            while(i<m) {
                count++;
                if(count==mid) {
                    start=nums1[i];
                }
                if(count==mid+1) {
                    end=nums1[i];
                    tag=1;
                    break;
                }
                i++;
            }

            if(tag==1) {
                if((m+n)%2==0) return ((double)start+end)/2;
                else return (double)end;
            } else {
                while(j<n) {
                    count++;
                    if(count==mid) {
                        start=nums2[j];
                    }
                    if(count==mid+1) {
                        end=nums2[j];
                        tag=1;
                        break;
                    }
                    j++;
                }
            }
        }

        if((m+n)%2==0) return ((double)start+end)/2;
        else return (double)end;
    }
};