Java算法(递归):两个不同长度的有序数组求第k小的元素(时间复杂度为:O(log(m1 + m2)))。

举报
泽宇-Li 发表于 2020/10/27 19:42:19 2020/10/27
【摘要】 首先看到时间复杂度要求就可以确定使用折半查找的思想。由于数组下标是从0开始计数,我们可以将k也定义为从0开始,那么原本题目中的第k小(k从1开始),在我们算法中就表示成第k-1小(k从0开始)。注意每次取mid1和mid2时,都应该考虑有没有越界情况的发生。public class ex_2 {public static void main(String[] args) {int[] arr...

首先看到时间复杂度要求就可以确定使用折半查找的思想。由于数组下标是从0开始计数,我们可以将k也定义为从0开始,那么原本题目中的第k小(k从1开始),在我们算法中就表示成第k-1小(k从0开始)。

注意每次取mid1和mid2时,都应该考虑有没有越界情况的发生。

public class ex_2 {public static void main(String[] args) {int[] arr1 = new int[]{1, 2, 3};int[] arr2 = new int[]{0, 4, 5, 6, 7, 8};System.out.println(findKth(arr1, arr2, 2));}public static int findKth(int[] arr1, int[] arr2, int k) {if (arr1 == null || arr1.length < 1) {return arr2[k - 1];}if (arr2 == null || arr2.length < 1) {return arr1[k - 1];}return findKth(arr1, 0, arr1.length - 1, arr2, 0, arr2.length - 1, k - 1);}public static int findKth(int[] arr1, int l1, int r1, int[] arr2, int l2, int r2, int k) {if (l1 > r1) {return arr2[l2 + k];}if (l2 > r2) {return arr1[l1 + k];}if (k == 0) {return Math.min(arr1[l1], arr2[l2]);}int mid1 = l1 + k / 2 < r1 ? l1 + k / 2 : r1;int mid2 = l2 + k / 2 < r2 ? l2 + k / 2 : r2;if (arr1[mid1] < arr2[mid2]) {return findKth(arr1, mid1 + 1, r1, arr2, l2, r2, k - (k / 2) - 1);} else if (arr1[mid1] > arr2[mid2]) {return findKth(arr1, l1, r1, arr2, mid2 + 1, r2, k - (k / 2) - 1);} else {return arr2[mid2];}}}

 


【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。