原题链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays/
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
示例 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
示例 3:
输入:nums1 = [0,0], nums2 = [0,0]
输出:0.00000
示例 4:
输入:nums1 = [], nums2 = [1]
输出:1.00000
示例 5:
输入:nums1 = [2], nums2 = []
输出:2.00000
题目分析
根据题目分析,需要把两个有序数组合并为一个有序数组然后输出其中位数,那么就需要把两个数组快速地合并,如果把数组b直接插入数组a,然后快速排序,那么时间复杂度就是,这样虽然能满足题目数据范围,但这显然浪费了很多时间,因为两个数组本来就是有序的,我们只需要做一些简单的操作就能把它们合并为一个新的有序数组。
双指针排序
双指针排序的原理就是使用两个变量,记录两个数组分别读取到什么位置了,每次比较两个指针所在的数字大小,把小的那个插入到新的数组里,然后把使用过的指针+1,反复进行这个操作,就可以得到一个新的有序数组,其复杂度是。
代码实现
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int nums3[] = new int[2000];
//初始化两个指针
int l=0;
int r=0;
//记录新数组保存位置
int ans =0;
while(l<nums1.length||r<nums2.length){
//初始化
int a=Integer.MAX_VALUE;
int b=Integer.MAX_VALUE;
if(l<nums1.length){
a=nums1[l];
}
if(r< nums2.length){
b=nums2[r];
}
//插入到新数组
if(a<=b){
nums3[ans++] = a;
l++;
}else{
nums3[ans++] = b;
r++;
}
// System.out.print(nums3[ans-1]);
}
//输出中位数
if((nums1.length+ nums2.length)%2==1){
return nums3[(nums1.length+ nums2.length)/2];
}else{
return (nums3[(nums1.length+ nums2.length)/2]+nums3[(nums1.length+ nums2.length)/2-1])/2.0;
}
}
}
二分查找
这道题还有一种解题思路,就是二分查找,我想了一下,做法如下:
1.找到两个数组中位数x,y
2.如果x<y,说明中位数在数组1后半段或数组2前半段
3.递归步骤1
后续有空会补上代码……