leetcode刷题4(寻找两个有序数组的中位数)
程序员文章站
2022-05-20 12:27:49
...
题目描述:
给定两个大小为 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 m=nums1.length;
int n=nums2.length;
int[] nums; //声明一个新数组用来存放两个数组合并后的情景
nums=new int[m+n];
int i=0;
int j=0;
int count=0;
if(m==0){ //若nums1数组为空
if(n%2==0){ //若nums2数组长度为偶数
return (nums2[n/2-1]+nums2[n/2])/2.0; //中位数为最中间两个数的平均
}else{
return nums2[n/2];
}
}
if(n==0){ //若nums2数组为空
if(m%2==0){
return (nums1[m/2-1]+nums1[m/2])/2.0;
}else{
return nums1[m/2];
}
}
while(count!=m+n){ //若新数组长度不为两数组合并后的长度
if(i==m){ //nums1已经遍历完
while(j!=n){
nums[count++]=nums2[j++]; //直接将nums2剩下的数放到新数组的后面
}
break;
}
if(j==n){ //若nums2已经遍历完
while(i!=m){
nums[count++]=nums1[i++];
}
break;
}
if(nums1[i]>nums2[j]){ //若nums1中的数大于nums2中的数
nums[count++]=nums2[j++]; //将小的数加入到新的数组
}else{
nums[count++]=nums1[i++];
}
}
if(count%2==0){ //若合并后的新数组的长度为偶数
return (nums[count/2-1]+nums[count/2])/2.0;
}else{
return nums[count/2];
}
}
}
上一篇: 类的继承
下一篇: 网页中< ?> 当中的代码表示的是什么