本类型博客中的各算法的时间复杂度分析均为博主自己推算,本类型博客也是博主自己刷LeetCode的自己的一些总结,因此个中错误可能较多,非常欢迎各位大神在博客下方评论,请不吝赐教
一、问题
- 输入:输入两个已经排好序(升序)的数组nums1和nums2,长度分别为m和n
- 输出:输出这两个数组合并之后的数组的中位数
- 条件:算法复杂度为 O(log(m+n))
二、输入输出示例
示例一:
- 输入:{1, 2}和{3}
- 输出:2,因为{1, 2, 3}的中位数为2
- 输入:{1, 2}和{3, 4]
- 输出:2.5,因为{1, 2 ,3 ,4}的中位数为(2+3)/2=2.5
三、解法
由于原始的两个数组就为有序数组,因此第一种思路就可以将两个数组合并,然后计算中位数即可。
解法一:合并两个数组然后计算
解题思路:可以利用归并排序中合并两个有序数组的思想将两个有序数组进行合并,由于只需要输出中位数,因此只需合并前(m+n)/2个数就可以了,后边的并不需要计算
package com.happy.leetcode.p4;
public class MedianofTwoSortedArraysV1 {
public static void main(String[] args) {
int[] nums1 = {1, 2};
int[] nums2 = {3, 4};
System.out.println(new MedianofTwoSortedArraysV1().findMedianSortedArrays(nums1, nums2));
}
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int count;
if((nums1.length+nums2.length)%2==1) {
count = (nums1.length+nums2.length+1)/2;
}else {
count = (nums1.length+nums2.length)/2+1;
}
int[] tmp = new int[count];
int p1 = 0;
int p2 = 0;
for(int i=0; i<count; i++) {
if(p1>=nums1.length || (p2<nums2.length && nums1[p1]>nums2[p2])) {
tmp[i] = nums2[p2++];
}else {
tmp[i] = nums1[p1++];
}
}
if((nums1.length+nums2.length)%2==1) {
return tmp[count-1];
}else {
return (tmp[count-2]+tmp[count-1])/2.0;
}
}
}
时间复杂度分析:需要获取前(m+n)/2个数,即需要遍历(m+n)/2次,因此算法时间复杂度为O(m+n)
【注意】该解法是该问题的一种解法,但是不符合题目对时间复杂度的要求