题目:
给定两个大小为 m 和 n 的正序(从小到大)数组 nums1
和 nums2
。请你找出并返回这两个正序数组的中位数。
示例 :
输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2
解析:
利用Linq的集合合并和排序得到从小到大的集合,通过二分法获取到中间的值,返回即可
代码:
public double FindMedianSortedArrays(int[] nums1, int[] nums2)
{
var newNums = nums1.Concat(nums2).OrderBy(x => x).ToList();
if(newNums.Count == 0)
{
return default;
}
var middle = (newNums.Count % 2 == 0 ? newNums.Skip(newNums.Count / 2 - 1).Take(2) : newNums.Skip(newNums.Count / 2).Take(1)).ToList();
return middle.Count > 1 ? middle.Sum() / 2.0 : middle.FirstOrDefault();
}
这里利用了Linq库图个方便,但是这个题的解题思想是一样的,只不过用Linq可以少些很多代码,但是真正运行效率来说的话 还是自己写快一些。自己写的话,将Concat、OrderBy、Skip、Take这些方法自己用循环实现一下即可。
还有一种解题思路就是不用合并两个集合,找两个集合的中位数就行。