《三战Leetcode》寻找有序数组的中位数

提示:

nums1.length == m

nums2.length == n

0 <= m <= 1000

0 <= n <= 1000

1 <= m + n <= 2000

-106 <= nums1[i], nums2[i] <= 106

进阶要求:你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗(注意这个要求,这个要求才是解决这道题目的关键)

三、题目解析

三、题目解析


这道题目很简单,就是从两个有序的数组中查询到它们的中文数,难点在于如何设计一个事件复杂度为O(log(m+n))的算法。 下面抽取题目关键信息来解读题目:

中位数:指顺序排序一组数据中居于中间位置的数值。 它分为两种情况,一是当数组长度为奇数时,正中间的数为中位数,二是当数组长度为偶数时,通常是将最中间的两个数相加取平均值作 为中位数,具体看下图:

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传

解法一:暴力破解


相信很多小伙伴一看到题目,脑海中就已经有了这种解题的思路:将两个数组合并起来 ,然后重新排序,再根据奇偶情况获取合并后的新数组的中位数。

1、解题代码:

/**

  • 方式一:时间复杂度和空间复杂度都为O(m+n)

  • @param nums1

  • @param nums2

  • @return

*/

public static double findMedianSortedArrays(int[] nums1, int[] nums2) {

int numLength = nums1.length + nums2.length;

// 定义合并后的集合

List list = new ArrayList();

// 使用for循环将元素添加到新的数组(方式一)

//for (int i : nums1) {

// list.add(i);

//}

//for (int i : nums2) {

// list.add(i);

//}

// 使用lambda表达式将数组合并到集合中(方式二)

list.addAll(Arrays.stream(nums1).boxed().collect(Collectors.toList()));

list.addAll(Arrays.stream(nums2).boxed().collect(Collectors.toList()));

// 使用结合工具类的排序方式对存放了两个数组的集合进行排序(sort底层的排序方式是:)

Collections.sort(list);

int middle = list.size() / 2;

// 两数组元素之和为偶数

if (numLength % 2 == 0) {

int leftMiddle = middle - 1;

double middleValue = (Double.valueOf(list.get(leftMiddle)) + Double.valueOf(list.get(middle))) / 2;

return middleValue;

} else {

return Double.valueOf(list.get(middle));

}

}

2、时间复杂度推导:

通过上面的代码,我们会发现随着输入规模的增大(即数组元素增多),程序需要花费执行时间处理的语句主要是在将[两个数组元素放入新的集合]以及对这个[新的集合进行排序]的过程。

将两个数组元素合并到一个数组执行函数可以使用函数:f(x)=m + n(m,n分别为两个数组的长度)表示,根据大O记法的推导可以得到时间复杂度为:O(m + n)

对新数组排序的Collections.sort()方法的最坏情况下时间复杂度为:O(n * log(n))。

因此,使用暴力破解的方式总的时间复杂度为:O(m+n) + O(n * log(n)),这个复杂度如果当数组长度变长后,效率是会比较低的,不推荐使用。

3、空间复杂度推导:

因为每次合并都需要申请一个新的集合来存放两个数组的元素,所以需要申请空间的函数可以表示为:f(x) = m + n,根据大O记法标准推导,可以得到空间复杂度为:O(m+ n)。

4、执行结果:

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传

解法二、双指针法


暴力破解法,当输入规模大的时候,效率极低,且没有满足题目的进阶要求,能否对时间复杂度和空间复杂度进一步优化呢?

答案是可以的。我们的目的是查询两个有序数组的中位数,在暴力破解中我们是通过申请新的数组集合来存放两个数组的值然后进行排序,最终得出结果,这一步是否真的需要呢?

答案是不需要,我们目的是查询中位数,中位数无非根据数组长度有奇偶两种情况,我们可以使用两个指针来指向对应的元素即可,这样我们就可以省略[申请新的数组空间]和对[新数组重新排序]的两个步骤,从而优化了时间复杂度和空间复杂度,下面来看看具体的代码。

public static double findMedianSortedArrays2(int[] nums1, int[] nums2) {

int num1Length = nums1.length;

int num2Length = nums2.length;

int sumLength = num1Length + num2Length;

// 第一个中位数指针

int preItem = 0;

// 第二个中位数指针(如果两个数组总长度为奇数,则直接返回该值即可)

int curItem = 0;

// 遍历的数组1的元素下标

int num1Index = 0;

// 遍历的数组2的元素下标

int num2Index = 0;

// 需要遍历的次数(查询中位数,并不需要将两个数组元素都遍历完)

int foreachTime = sumLength / 2;

for (int i = 0; i <= foreachTime; i++) {

// 保证preItem总是在curItem前面

preItem = curItem;

// num2Index >= num2Length需要放在nums1[num1Index] < nums2[num2Index]前面,否则会出现下标越界

// nums1[num1Index] < nums2[num2Index]的目的就是为了按照模拟从小到大的顺序循环两个数组的元素

if (num1Index < num1Length && (num2Index >= num2Length || nums1[num1Index] < nums2[num2Index])) {

curItem = nums1[num1Index++];

} else {

curItem = nums2[num2Index++];

}

}

// 如果是偶数,则取中间两个元素的平均值

if (sumLength % 2 == 0) {

return Double.valueOf(preItem + curItem) / 2;

}

return curItem;

}

1、时间复杂度推导

根据上面代码可知,随着输入规模的增大(即数组元素增多),程序需要执行花费时间处理的语句主要是在for循环中(for循环又只跟两个数组的长度相关),可以用函数表示为:f(n) = m + n,根据大O记法规则推断,该方式的时间复杂度为:O(m + n)。

2、空间复杂度推导

根据上面解题代码可知,使用双指针法只需要申请两个指针和一些存储长度、下标的变量对应的空间,并且这些空间并不会随着问题规模的增大(即数组元素的增多)而变化,因此该算法的空间复杂度为:O(1)

3、执行结果

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传

4、小结

通过定义双指针模拟指向中位数,我们去除了不必要的空间申请和重新排序,进一步优化了算法的时间复杂度和空间复杂度,在平常的业务中,如果遇到相似的业务要求,可以优先考虑维护指针的方式来避免不必要的空间申请。

解法三、二分查找法


通过双指针,我们将算法的时间复杂度降低到了O(m +n),但是依然没有达到题目中O(log(m + n))的要求。如何满足这个要求呢,现在好像没有一个比较清晰的思路,这时候不妨再反过来思考下题目的要求。

题目中要求时间复杂度需要达到O(log(m + n)),回想下我们之前接触到的算法中,有没有与log(对数)相关的东西,没错,比较常见的就是二分法,每次循环都排除n/2的元素,最终得出结果,下面来看看这个题目如何提取成二分法的形式。

题目最终结果是要求中位数,中位数又分为奇偶情况,那我们就可以将抽象求中位数成求有序数组中的第k小数,其中k就是对应的中位数(即 (m + n) /2,或者(m+n)/2 +1),这样我们就可以对k进行二分查找法找到符合条件的数值。

1、求解第k小数的思路

假设存在数组A和数组B,它们的中位数为k,此时要求k的值,则可以通过二分法,即每轮都对A[k/2]和B[k/2]进行比较(注意:这里的k是表示第几个,如果转换成数组对应的元素的话需要减去1),如果A[k/2]>=B[k/2],则B[k/2]和B[k/2]之前的元素就可排除掉,原因如下:

当A[k/2]>=B[k/2]时,在A数组中比A[k/2]元素小的值有k/2-1个,在数组B中比B[k/2]小的有k-1个,即使A数组中A[k/2]之前的所有元素都比B[k/2]元素小,那总的个数也是等于:k/2-1 + k/2-1 = k-2个,所以B[k/2]最多只能是k-1小的数,而不是第k小数,所以B[k/2]之前的数组更不可能是第k小数,故B[k/2]及之前的元素可以排除掉。反之亦然。

因为排除掉的元素一定位于数组的前面(数组是有序的),所以每轮之后k的值也需要减去排除掉的元素的个数,然后再进行下一轮的第k小数查询,步骤依次类推。

通过上面的思路整理,我们可以看出此处使用了递归的思想,递归的出口则是当某个数组长度为了0时(此时中位数就是可以取不为0的数组中的值即可)或者是k=1(即求第1个小数,此时中位数则取两个数组中起始下标对应值最小的元素)时,需要注意的是:因为k/2的值可能大于数组的长度,所以每次比较 min(k/2,len(数组) 对应的数字,把小的那个对应的数组的数字排除,将两个新数组进入递归。

2、图解步骤讲解

假设存在数组A元素有[1,2,3]和数组B元素有[1,2,3,4,5,6],因为k=(A数组长度 + B数组长度 + 1)/2 = 5,则根据第k小数的方式查询中位数的步骤如下:

第一轮循环:

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传

第二轮循环:

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传

第三轮循环:

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传

第四轮循环:

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传

3、代码讲解

public double findMedianSortedArrays(int[] nums1, int[] nums2) {

int n = nums1.length;

int m = nums2.length;

//奇数时k值,除法是向下取整,因此需要+1

int left = (n + m + 1) / 2;

// 偶数时取两个中间值的平均值做中位数

int right = (n + m + 2) / 2;

//将偶数和奇数的情况区分

if((n + m) % 2 == 0){

return (getMedian(nums1, 0, n - 1, nums2, 0, m - 1, left) + getMedian(nums1, 0, n - 1, nums2, 0, m - 1, right)) * 0.5;

} else {

return getMedian(nums1, 0, n - 1, nums2, 0, m - 1, left);

}

}

public static double getMedian(int[] num1, int num1StartIndex, int num1EndIndex, int[] num2, int num2StartIndex, int num2EndIndex, int k) {

// 获取需要遍历数组的长度(不能直接用num1.length获取,这样会导致数组的长度一致不会变动)

int num1Length = num1EndIndex - num1StartIndex + 1;

int num2Length = num2EndIndex - num2StartIndex + 1;

// 总长度

int totalLength = num1Length + num2Length;

// 如果num1长度大于num2,则进行交换位置,保证num1数组为包含元素最少的数组

if (num1Length > num2Length) {

return getMedian(num2, num2StartIndex, num2EndIndex, num1, num1StartIndex, num1EndIndex, k);

}

// 如果最短的数组长度为0,则中位数获取剩下还有元素的数组

if (num1Length == 0) {

return num2[num2StartIndex + k - 1];

}

// 如果k为1(表示查询第一小的数值),表示数组第一个元素为中位数,则取两个数组中位数中最小的值

if (k == 1) {

return Math.min(num1[num1StartIndex], num2[num2StartIndex]);

}

// 计算两个数组起始下标的位置

// 疑问:为什么比较的两个值需要添加num1StartIndex和num2StartIndex,原因是比较的是k/2的元素,但是每轮递归后

// k需要减去上一轮排除的元素的个数

// 减去1是因为前面计算的都是长度,但是offset对应的是数组下标

// Math.min是为了防止数组长度比k/2长度小而导致数组越界的问题,使用它表示如果数组长度小于k/2时,则直接将数组坐标指到最后一个

int num1CompareItem = num1StartIndex + Math.min(num1Length, k / 2) - 1;

int num2CompareItem = num2StartIndex + Math.min(num2Length, k / 2) - 1;

// 判断两个数组中位数的值

if (num1[num1CompareItem] > num2[num2CompareItem]) {

return getMedian(num1, num1StartIndex, num1EndIndex, num2, num2CompareItem + 1, num2EndIndex, k - (num2CompareItem - num2StartIndex+ 1));

} else {

return getMedian(num1, num1CompareItem + 1, num1EndIndex, num2, num2StartIndex, num2EndIndex, k - (num1CompareItem -num1StartIndex + 1));

}

}

4、执行结果

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传

5、时间复杂度推导

自我介绍一下,小编13年上海交大毕业,曾经在小公司待过,也去过华为、OPPO等大厂,18年进入阿里一直到现在。

深知大多数Java工程师,想要提升技能,往往是自己摸索成长或者是报班学习,但对于培训机构动则几千的学费,着实压力不小。自己不成体系的自学效果低效又漫长,而且极易碰到天花板技术停滞不前!

因此收集整理了一份《2024年Java开发全套学习资料》,初衷也很简单,就是希望能够帮助到想自学提升又不知道该从何学起的朋友,同时减轻大家的负担。img

既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,基本涵盖了95%以上Java开发知识点,真正体系化!

由于文件比较大,这里只是将部分目录截图出来,每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频,并且会持续更新!

如果你觉得这些内容对你有帮助,可以扫码获取!!(备注Java获取)

img

总结

这份面试题几乎包含了他在一年内遇到的所有面试题以及答案,甚至包括面试中的细节对话以及语录,可谓是细节到极致,甚至简历优化和怎么投简历更容易得到面试机会也包括在内!也包括教你怎么去获得一些大厂,比如阿里,腾讯的内推名额!

某位名人说过成功是靠99%的汗水和1%的机遇得到的,而你想获得那1%的机遇你首先就得付出99%的汗水!你只有朝着你的目标一步一步坚持不懈的走下去你才能有机会获得成功!

成功只会留给那些有准备的人!

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取!
础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,基本涵盖了95%以上Java开发知识点,真正体系化!**

由于文件比较大,这里只是将部分目录截图出来,每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频,并且会持续更新!

如果你觉得这些内容对你有帮助,可以扫码获取!!(备注Java获取)

img

总结

这份面试题几乎包含了他在一年内遇到的所有面试题以及答案,甚至包括面试中的细节对话以及语录,可谓是细节到极致,甚至简历优化和怎么投简历更容易得到面试机会也包括在内!也包括教你怎么去获得一些大厂,比如阿里,腾讯的内推名额!

某位名人说过成功是靠99%的汗水和1%的机遇得到的,而你想获得那1%的机遇你首先就得付出99%的汗水!你只有朝着你的目标一步一步坚持不懈的走下去你才能有机会获得成功!

成功只会留给那些有准备的人!

[外链图片转存中…(img-qvYvmZu0-1713515617029)]

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取!

  • 22
    点赞
  • 29
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值