寻找两个正序数组的中位数(暴力解法,双指针同时遍历,寻找第K小的数)

暴力解法:
将两个有序数组合并起来,然后取中位数。

public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int l1 = nums1.length;
        int l2 = nums2.length;
        if (l1 == 0) {
            return ((l2 % 2) == 0 ? ((nums2[l2 / 2] + nums2[l2 / 2 - 1]) / 2.0) : nums2[l2 / 2]);
        }
        if (l2 == 0) {
            return ((l1 % 2) == 0 ? ((nums1[l1 / 2] + nums1[l1 / 2 - 1]) / 2.0) : nums1[l1 / 2]);
        }

        int nums[] = new int[l1 + l2];
        int count = 0;
        int i = 0;
        int j = 0;
        while (count != nums.length) {
            if (i == l1) {
                while (j != l2) {
                    nums[count++] = nums2[j++];
                }
                break;
            }
            if (j == l2) {
                while (i != l1) {
                    nums[count++] = nums1[i++];
                }
                break;
            }

            if (nums1[i] < nums2[j]) {
                nums[count++] = nums1[i++];
            } else {
                nums[count++] = nums2[j++];
            }
        }
        if (count % 2 == 0) {
            return (nums[count / 2 - 1] + nums[count / 2]) / 2.0;
        } else {
            return nums[count / 2];
        }
    }

同样因为数组长度已知,中位数下标已知,通过两个指针同时遍历两个数组也可获取中位数。

第K小:
要找中位数想法等同于查找第中位数K小的数(长度为偶数时还需求第K-1小的数并除2)
想法是将K分为两部分(对应有两个数组),将类似滑动窗口左边界的处理,首先将两数组左边界改为当前位置+(K/2)(注意此值可能下标越界,若越界则指向最后一个下标即可),若数组1的左边界<数组2的左边界,那么数组1左边界以左包括边界都不可能是第K小的数,那么就把数组1左边界右移一个位置,K也需要减去不可能的数;
重复该操作,直到①有数组遍历完,则结果为另一个数组当前下标加K-1②当K为1,结果为两个数组当前位置较小数。

public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int l1 = nums1.length;
        int l2 = nums2.length;
        int whole = l1 + l2;
        if (whole % 2 == 0) {
            return ((getK(nums1, nums2, whole / 2 + 1) + getK(nums1, nums2, whole / 2)) / 2.0);
        } else {
            return (getK(nums1, nums2, whole / 2 + 1));
        }
    }

    int getK(int[] nums1, int[] nums2, int k) {
        int index1 = 0;
        int index2 = 0;

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

            int half = k / 2;
            int new1 = Math.min(nums1.length, index1 + half) - 1;
            int new2 = Math.min(nums2.length, index2 + half) - 1;
            if (nums1[new1] > nums2[new2]) {
                k -= (new2 - index2 + 1);
                index2 = new2 + 1;
            } else {
                k -= (new1 - index1 + 1);
                index1 = new1 + 1;
            }
        }
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
### 回答1: 你可以使用归并排序的思想来寻找两个数组中位数。具体来说,你可以比较两个数组的首个字,将较小的字从原数组中删除,并将它放入一个临时数组。然后,你可以继续比较两个数组的首个字,将较小的字从原数组中删除,并将它放入临时数组。你可以重复这个过程,直到其中一个数组为空。最后,你可以将另一个数组中剩余的字放入临时数组中。 最后,你可以计算临时数组中位数。如果临时数组的大小为奇,则中位数就是临时数组的第 (n+1)/2 个字(n 是临时数组的大小)。如果临时数组的大小为偶,则中位数就是临时数组的第 n/2 个字和第 (n/2)+1 个字的平均。 示例代码(使用 C 语言): double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){ int i = 0, j = 0, k = 0; int m = nums1Size, n = nums2Size; int *merged = (int *)malloc((m + n) * sizeof(int)); while (i < m && j < n) { if (nums1[i] < nums2[j]) { merged[k] = nums1[i]; i++; } else { merged[k] = nums2[j]; j++; } k++; } while (i < m) { merged ### 回答2: 首先,我们可以将两个数组合并成一个有序数组,然后根据数组长度的奇偶性来确定中位数的位置。 我们定义两个指针i和j,分别指向nums1和nums2的开头,同时定义两个变量prev和curr,初始化为0。prev表示上一个元素,curr表示当前元素。 我们使用while循环遍历数组,同时判断curr的值是否大于等于中位数的位置。如果是,就停止循环。 在循环中,我们比较nums1[i]和nums2[j]的大小,将较小的值赋给prev,并将较大的值赋给curr。然后移动指针i或j,具体移动哪个指针取决于哪个值较小。 当curr的值达到中位数的位置时,我们根据数组长度的奇偶性来确定中位数。如果数组长度为奇,那么中位数就是curr的值。如果数组长度为偶,那么中位数就是prev和curr的平均值。 最后,我们返回中位数即可。 时间复杂度分析:由于是有序数组,将两个数组合并成一个数组的时间复杂度为O(m+n)。而循环遍历数组的时间复杂度为O((m+n)/2),算法的时间复杂度为O(log(m*n)),满足题目要求。 代码示例: ```c double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){ int totalSize = nums1Size + nums2Size; int mid = (totalSize - 1) / 2; int i = 0, j = 0; int prev = 0, curr = 0; while (i < nums1Size && j < nums2Size && curr <= mid) { prev = curr; if (nums1[i] < nums2[j]) { curr = nums1[i++]; } else { curr = nums2[j++]; } } while (i < nums1Size && curr <= mid) { prev = curr; curr = nums1[i++]; } while (j < nums2Size && curr <= mid) { prev = curr; curr = nums2[j++]; } if (totalSize % 2 == 0) { return (prev + curr) / 2.0; } else { return curr; } } ``` 以上是一种解法,还有其他解法可以使用二分查找来寻找中位数。总的来说,这是一个经典的问题,可以有多种解法

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

魔幻音

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值