一天一道算法题(2020-10-08)

题目:

给定两个大小为 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这些方法自己用循环实现一下即可。

还有一种解题思路就是不用合并两个集合,找两个集合的中位数就行。


版权声明:本文为wengpiwang原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。