首页 > 代码库 > leetcode的题4:给定两个有序数组,求两个数组的中间值。

leetcode的题4:给定两个有序数组,求两个数组的中间值。

我的思路是,既然是有序数组,就可以按照归并排序法的思路,按照最后的归并过程。

建立一个新的数组,并对两个数组及归并数组目前的位置分别编号i,j,k。

将两个数组中较小的值推入i的位置,然后将被提取数据的数组及归并数组索引+1。

这个过程中需要考虑的问题是如果其中一个数组已经全部提取完成应如何处置。

以下是我的代码:

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        size_t n = nums1.size(),m=nums2.size();
        vector<int> combine;
        int i=0,j=0,k=0;
        for(;i<m+n;++i){
            if(j<n&&k<m)
                combine.push_back(nums1[j]<nums2[k]?nums1[j++]:nums2[k++]);
            else
                combine.push_back(j==n?nums2[k++]:nums1[j++]);
        }
        if((m+n)%2==1)
            return combine[(m+n-1)/2];
        else
            return (combine[(m+n-1)/2]+combine[(m+n-1)/2+1])/2.0;
    }
};

 

leetcode的题4:给定两个有序数组,求两个数组的中间值。