题目:寻找两个正序数组的中位数
描述:
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数 。
算法的时间复杂度应该为 O(log (m+n)) 。
示例 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
解法:先将两个数组进行排序,排序之后判断其元素个数是奇数还是偶数,如果是奇数,则直接输出中间元素,若是偶数,则取中间两位数之和除以2。
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int[] nums3 = new int[nums1.length+nums2.length]; //创建新数组,来存放排序结果
int i = 0,j = 0,k = 0; //i指向nums1数组,j指向nums2数组,k指向nums3数组
double result = 0; //存放中间数
//对两个数组排序
while(i < nums1.length && j < nums2.length){
if(nums1[i] < nums2[j]){
nums3[k++] = nums1[i++];
}else{
nums3[k++] = nums2[j++];
}
}
//如果num1数组没放完,则剩下元素全部放进新数组
while(i < nums1.length){
nums3[k++] = nums1[i++];
}
//如果num2数组没放完,则剩下元素全部放进新数组
while(j < nums2.length){
nums3[k++] = nums2[j++];
}
//k记录了数组中的元素个数,来对k进行判断
if(k % 2 == 0){
result = (nums3[k/2 - 1] + nums3[k/2])*1.0 / 2;
}else{
result = nums3[k/2];
}
return result;
}
}
总结:最后该算法的时间耗费为1ms,在内存方面比较差,消耗了42,3MB,在这里埋个坑,看以后能不能在内存方面做一些优化。