題目
給定兩個大小分別為 m 和 n 的正序(從小到大)數(shù)組 nums1 和 nums2。請你找出并返回這兩個正序數(shù)組的 中位數(shù) 。
算法的時間復(fù)雜度應(yīng)該為 O(log (m+n)) 。
示例 1:
輸入:nums1 = [1,3], nums2 = [2]
輸出:2.00000
解釋:合并數(shù)組 = [1,2,3] ,中位數(shù) 2
示例 2:
輸入:nums1 = [1,2], nums2 = [3,4]
輸出:2.50000
解釋:合并數(shù)組 = [1,2,3,4] ,中位數(shù) (2 + 3) / 2 = 2.5
來源:力扣(LeetCode)
鏈接:https://leetcode.cn/problems/median-of-two-sorted-arrays
代碼
- 時間復(fù)雜度為 O(m+n)
class Solution {
double findMedianSortedArrays(List<int> nums1, List<int> nums2) {
List<int> list = nums1 + nums2;
if (list.isEmpty) return 0;
if (list.length == 1) return list.first.toDouble();
if (list.length == 2) return (list.first + list.last) / 2;
list.sort();
int index = list.length ~/ 2;
if (list.length % 2 == 0) {
return (list[index - 1] + list[index]) / 2;
} else {
return list[index].toDouble();
}
}
}
- 時間復(fù)雜度為 O(log (m+n))
這個太難了嫉称,暫時不會??