letcode 两个排序数组的中位数
程序员文章站
2024-03-16 12:51:16
...
题目描述:
给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。
请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。
你可以假设 nums1 和 nums2 不同时为空。
示例 1:
nums1 = [1, 3]
nums2 = [2]
中位数是 2.0
示例 2:
nums1 = [1, 2]
nums2 = [3, 4]
中位数是 (2 + 3)/2 = 2.5
思路:这个题只需归并两个数组就好了,之后找中位数就轻而易举.
实现:
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int[] num = merge(nums1,nums2);
int N = num.length;
double result=0;
if(N%2==0) {
result = (num[N/2]+num[N/2-1])/2.0;
}else {
result=num[N/2];
}
return result;
}
//用于归并数组,从小到大
public int[] merge(int[] nums1, int[] nums2) {
int[] result = new int[nums1.length+nums2.length];
if(nums1.length==0) return nums2;
if(nums2.length==0) return nums1;
int i=0;int j=0;
int temp =0;
while(i<nums1.length||j<nums2.length) {
//归并小的那一个
if(nums1[i]>nums2[j]) {
result[temp++] = nums2[j++];
}else {
result[temp++] = nums1[i++];
}
//如果一个数组已经归并完归并另一个
while(i>=nums1.length) {
if(j>=nums2.length) break;
result[temp++] = nums2[j++];
}
while(j>=nums2.length) {
if(i>=nums1.length) break;
result[temp++] = nums1[i++];
}
}
return result;
}
}
上一篇: 算法系列------两数之和