Leetcode解题报告:4. Median of Two Sorted Arrays

来源:互联网 发布:分销平台软件 编辑:程序博客网 时间:2024/06/13 13:47

题意:

There are two sorted arrays nums1 and nums2 of size m and n respectively.

Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

难度:Hard

解题思路: 看到题目第一个跳出来的想法是把两个数组合并成一个数组,然后这样时间复杂度是O(m+n)。这道题要求在O(log(m+n))的复杂度内完成,显然合并的思路不合适。我们要求中间的那个数,设两个长度为m,n的数组,(m+n)为奇数的时候,中位数就是两个合并以后第(m+n)/2那个数,否则就是第(m+n)/2和第(m+n)/2+1个数的中位数。

那么可以考虑用分治法,由于两个数组都已经各自有序了,不失一般性,我们可以假设nums1为长度较小的那个数组。我们比较第一个数组的第k1个数Nk1和第二个数组第k2个数Nk2,且k1+k2=k,当K/2小于nums1.size()的时候,取k1=k/2,否则k1=nums1.size()。如果Nk1=Nk2,那就说明它们就等于两个数组中第K个数。如果Nk1<Nk2,那么第一个数组的第1到第K1个数中不会有所要找的那个第K大的数,所以问题可以缩小为求以第k1个数为轴,第一个数组的右半部分和第二个数组全部中的第k-k1个数是多少。同理,如果Nk1>Nk2,问题就可以被缩小为第一个数组全部和第二个数组中以第k2个数为轴右边部分所组成的两个数组中第k-k2个数是多少。 如果有一个数组的大小为0,那么第k个数就是另外一个数组的第k个数,由于我们之前已经假设nums1为长度较小的那个数组,所以返回nums2[start+k-1];  

class Solution {public:double helper(vector<int>& nums1,vector<int> &nums2,int s1,int l1,int s2,int l2,int k){if(k==1){    int ans;    if(l1!=0&&l2!=0)    ans=min(nums1[s1],nums2[s2]);else if(l1==0)    ans=nums2[s2];else     ans=nums1[s1];    return ans;}if(l1>l2){return helper(nums2,nums1,s2,l2,s1,l1,k);}if(l1==0){return nums2[s2+k-1];}int k1=min(k/2,l1);int k2=k-k1;if(nums1[s1+k1-1]<nums2[s2+k2-1]){return helper(nums1,nums2,s1+k1,l1-k1,s2,l2,k-k1);}else if(nums1[s1+k1-1]>nums2[s2+k2-1]){return helper(nums1,nums2,s1,l1,s2+k2,l2-k2,k-k2);}else {    return nums2[s2+k2-1];}}double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {int l1=nums1.size();int l2=nums2.size();if((l1+l2)%2!=0){return helper(nums1,nums2,0,l1,0,l2,(l1+l2)/2+1);}else {return (helper(nums1,nums2,0,l1,0,l2,(l1+l2)/2)+helper(nums1,nums2,0,l1,0,l2,(l1+l2)/2+1))/2;}        }};

以上参考:《算法概述》、林老师算法设计与应用基础课堂讲解。

0 0
原创粉丝点击