题目
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
思路
新建一个数组,用来保存题目中给出的两个数组合并后的结果,在合并的时候需要考虑两个数组的指针是否完全遍历的情况,有可能会出现短数组已经遍历完了,另一个数组还没有遍历完。
然后考虑数组大小是偶数还是奇数,奇数数组直接返回中间位置,偶数元素个数的数组需要额外考虑
代码
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int []newarr=new int [nums1.length+nums2.length];
int a=0;int b=0;int i=0;
while(a<nums1.length&&b<nums2.length){
if(nums1[a]>nums2[b]){
newarr[i]=nums2[b];
b++;
i++;
}else{
newarr[i]=nums1[a];
a++;
i++;
}
}
if(a<nums1.length){
for(int k=a;k<nums1.length;k++){
newarr[i]=nums1[k];
i++;
}
}
if(b<nums2.length){
for(int k=b;k<nums2.length;k++){
newarr[i]=nums2[k];
i++;
}
}
int mid = newarr.length / 2;
if(newarr.length % 2 == 0) {
return (double)(newarr[mid - 1] + newarr[mid]) / 2;
} else {
return newarr[mid];
}
}
}