题目描述
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
解题思路
将两个有序数组归并为一个有序数组,直接获取对应下标的元素即可。
解题代码
// 归并两个数组
public int[] merge(int[] nums1, int[] nums2) {
int[] total = new int[nums1.length + nums2.length];
// i: nums1的索引,j: nums2的索引, k: total的索引
int i = 0, j = 0, k = 0;
// 遍历并比较两个数组的元素并且将较小的元素放入total数组中
while (i < nums1.length && j < nums2.length) {
// 比较大小
if (nums1[i] < nums2[j]) {
total[k++] = nums1[i++];
} else {
total[k++] = nums2[j++];
}
}
// 如果nums1数组还有元素
while (i < nums1.length) {
total[k++] = nums1[i++];
}
// 如果nums2数组还有元素
while (j < nums2.length) {
total[k++] = nums2[j++];
}
return total;
}
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int[] total = merge(nums1, nums2);
int len = total.length;
if (total.length % 2 == 0) {
return (total[len / 2 - 1] + total[len / 2]) / 2.0;
} else {
return total[len / 2] * 1.0;
}
}
题目来源
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays/