其他分享
首页 > 其他分享> > leedCode 4寻找两个正序数组的中位数

leedCode 4寻找两个正序数组的中位数

作者:互联网

给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
在这里插入图片描述

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
//        int medium = 0;
//        ArrayList<Integer> arrayList=new ArrayList<>();
//        for (int i = 0; i < nums1.length; i++) {
//            arrayList.get(i);
//        }
//        if (nums1.length >= nums2.length) {
//            for (int i = 0; i < nums1.length; i++) {
//                for (int j = 0; j <nums1.length; j++) {
//                    if (nums1[i]>=nums2[j]){
//                        arrayList.get(j);
//                    }
//                }
//                medium= (int) arrayList.get((nums1.length + nums2.length) / 2);
//            }
//        }
//        else {
//            for (int i = 0; i < nums2.length; i++) {
//                for (int j = 0; j < nums2.length; j++) {
//                    if (nums1[i]>=nums2[j]){
//                        arrayList.get(j);
//                    }
//                }
//            }
//            medium= (int) arrayList.get((nums1.length + nums2.length) / 2);
//        }
//        return medium;
        //假设nums1[1,2],nums2[3,4]
        //num:1 2 3 4  len=4
        //pos:0 1 2 3 ->中位数:pos=(num(1)+num(2)/2)=([num(len-1)/2]+[num(len-1)/2+1])/2
        //数组的长度是奇数:mid_pos:(len-1)/2
        //数组的长度是偶数:mid_pos:(len-1)/2和(len-1)/2+1
        int len=nums1.length+nums2.length;
        //index-合并之后的数组下标,index1-nums1数组的的下标,index2-nums2数组的下标
        int index=0;int index1=0;int index2=0;
        double ans=0.0;//初始化结果
        //中位数开始的下标
        int pos=(len-1)/2;
        boolean enevNum=len%2==0;
        while (index<len){
            //获取nums1数组
            Integer num1 = index1 < nums1.length ? nums1[index1] : null;
            //获取nums2数组
            Integer num2 = index2 < nums2.length ? nums2[index2] : null;
            //比较一下num1和num2的大小
            int num;
            if (num1 != null && num2 != null) {//说明两个数组并没有遍历完
                num = num1 < num2 ? nums1[index++] : nums2[index2++];
            } else {
                num = num1 == null ? nums1[index++] : nums2[index1++];
            }
            if (index == pos) {
                if (enevNum) {
                    //nums1+nums2的长度是偶数
                    ans += num;
                } else {
                    //nums1+nums2的长度是奇数
                    ans += num;
                    break;
                }
            } else if (index == pos + 1) {
                //代码能走到这里说明合并之后的数组长度一定是偶数
                ans += num*2;
            }
            //遍历合并之后的数组下标+1
            index++;
        }
        return ans/2;
    }
}

标签:正序,int,leedCode,中位数,length,num,数组,nums1,nums2
来源: https://blog.csdn.net/tutouxiaocaiji/article/details/120420747