Median of Two Sorted Arrays

来源:互联网 发布:网络本科学历可以考研 编辑:程序博客网 时间:2024/06/16 19:03

**question:**There are two sorted arrays nums1 and nums2 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)).

Example 1:
nums1 = [1, 3]
nums2 = [2]

The median is 2.0
Example 2:
nums1 = [1, 2]
nums2 = [3, 4]

The median is (2 + 3)/2 = 2.5
个人思路:寻找两个有序数组中所有元素的中位数,没有其他捷径,只能在对组合后的数组进行排序之后才能找到中位数。所以重心放在这(m+n)个元素的排序上面。唯一区别的是,这是两个有序序列,所以当任意nums1中的元素Xi大于nums2中的某元素Yj时,则有,对任意m>i,nums1中的元素Xm必定都大于Yj。按照这个思路,只需要记录下上一次的比较结果,则可以减少比较的次数。

public double findMedianSortedArrays(int[] nums1, int[] nums2) {        int len1 = nums1.length;        int len2 = nums2.length;        int temp2=0;  //记录nums2中的上次插入位置        ArrayList<Integer> res = new ArrayList<Integer>();        //将nums1向nums2中插入        for(int i=0;i<len1;i++){            for(int j=temp2;j<len2;j++){                if(nums1[i]<nums2[j]) {                    res.add(nums1[i]);                    temp2=j;                    break;                }else{                    res.add(nums2[j]);                    if(j>=len2-1) {                        temp2 = len2;                    }                }            }            if(temp2>=len2){                res.add(nums1[i]);            }        }        if(temp2<len2){//nums2中还有元素未被插入            for(int i=temp2;i<len2;i++){                res.add(nums2[i]);            }        }        if(0==(len1+len2)%2)            return (res.get((len1+len2)/2-1)+res.get((len1+len2)/2))/2.0;        else            return res.get((len1+len2)/2);    }
0 0
原创粉丝点击