LeetCode 4.寻找两个正序数组的中位数二分查找每日一题

@小红花

已于 2022-04-16 08:28:51 修改

阅读量1.4k

点赞数 18

分类专栏： LeetCode 文章标签： c语言职场和发展 java 算法 leetcode

于 2022-04-15 21:00:23 首次发布

本文链接：https://blog.csdn.net/lishifu_/article/details/124202634

版权

LeetCode 专栏收录该内容

84 篇文章 0 订阅

订阅专栏

1、问题描述

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。

算法的时间复杂度应该为 O(log (m+n)) 。

示例 1：

输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2
示例 2：

输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/median-of-two-sorted-arrays

2、Java Burst Force 合并数组

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int[] arr = new int[nums1.length + nums2.length];
        for(int i = 0;i < nums1.length;i++){
            arr[i] = nums1[i];
        }
        int index = nums1.length;
        for(int i : nums2){
            arr[index++] = i;
        }
        Arrays.sort(arr);
        
        if(arr.length % 2 == 1){
            return 1.0 * arr[arr.length / 2];
        }else {
            return (1.0 * arr[arr.length / 2] + arr[arr.length / 2 - 1]) / 2;
        }
    }
}

3、Java 二分查找

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int length1 = nums1.length;
        int length2 = nums2.length;
        int total = length1 + length2;
        if(total % 2 == 1){
            int middle = total / 2;
            double ans = getMedian(nums1,nums2,middle + 1);
            return ans;
        }else {
            int middle = total / 2;
            double ans = (getMedian(nums1,nums2,middle) + getMedian(nums1,nums2,middle + 1)) / 2.0;
            return ans;
        }
    }
    public double getMedian(int[] nums1,int[] nums2,int k){
        int length1 = nums1.length;
        int length2 = nums2.length;
        //以当前下标为基点以及前面的数字都不是第k个数，过滤k - 1个数就可以了
        int index1 = 0;
        int index2 = 0;

        while(true){
            //边界
            if(index1 == length1){
                return nums2[index2 + k - 1];
            }
            if(index2 == length2){
                return nums1[index1 + k - 1];
            }
            
            if(k == 1){
                return Math.min(nums1[index1],nums2[index2]);
            }

            int newIndex1 = Math.min(index1 + k / 2,length1) - 1;
            int newIndex2 = Math.min(index2 + k / 2,length2) - 1;

            int n1 = nums1[newIndex1];
            int n2 = nums2[newIndex2];

            if(n1 <= n2){
                k -= (newIndex1 - index1 + 1); // k -= k / 2;
                index1 = newIndex1 + 1;
            }else {
                k -= (newIndex2 - index2 + 1); // k -= k / 2;
                index2 = newIndex2 + 1;
            }
        }
    }

}

4、C语言二分查找

double getMedian(int* nums1, int nums1Size, int* nums2, int nums2Size,int k){
    //以当前下标为基点以及前面的数字都不是第k个数，过滤k - 1个数就可以了
    int index1 = 0;
    int index2 = 0;

    while(true){
        //边界
        if(index1 == nums1Size){
            return nums2[index2 + k - 1];
        }
        if(index2 == nums2Size){
            return nums1[index1 + k - 1];
        }
        
        if(k == 1){
            return nums1[index1] < nums2[index2] ? nums1[index1] : nums2[index2];
        }

        int newIndex1 = index1 + k / 2 < nums1Size ? index1 + k / 2 - 1 : nums1Size - 1;
        int newIndex2 = index2 + k / 2 < nums2Size ? index2 + k / 2 - 1 : nums2Size - 1;

        int n1 = nums1[newIndex1];
        int n2 = nums2[newIndex2];

        if(n1 <= n2){
            k -= (newIndex1 - index1 + 1); // k -= k / 2;
            index1 = newIndex1 + 1;
        }else {
            k -= (newIndex2 - index2 + 1); // k -= k / 2;
            index2 = newIndex2 + 1;
        }
    }
}
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){
    int total = nums1Size + nums2Size;
    if(total % 2){
        int middle = total / 2;
        double ans = getMedian(nums1,nums1Size,nums2,nums2Size,middle + 1);
        return ans;
    }else {
        int middle = total / 2;
        double ans = (getMedian(nums1,nums1Size,nums2,nums2Size,middle) + 
        getMedian(nums1,nums1Size,nums2,nums2Size,middle + 1)) / 2.0;
        return ans;
    }
}