LeetCode - 4. Median of Two Sorted Arrays

简介: 4. Median of Two Sorted Arrays  Problem's Link  ---------------------------------------------------------------------------- Mean:  给定两个数组,求这两个数组的中位数.

4. Median of Two Sorted Arrays 

Problem's Link

 ----------------------------------------------------------------------------

Mean: 

给定两个数组,求这两个数组的中位数.(要求时间复杂度为O(log(n+m))

analyse:

一开始用归并为一个数组的方法做了一下也AC了,看来lc的时间还是给的很宽的.

将本题转化为求第k大数就简单多了,其中求第k大数使用类似二分的方法来实现,从而将时间复杂度降到O(log(n+m)).

Time complexity: O(log(n+m)

 

view code

/**
* -----------------------------------------------------------------
* Copyright (c) 2016 crazyacking.All rights reserved.
* -----------------------------------------------------------------
*       Author: crazyacking
*       Date  : 2016-02-03-12.07
*/
#include <queue>
#include <cstdio>
#include <set>
#include <string>
#include <stack>
#include <cmath>
#include <climits>
#include <map>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long( LL);
typedef unsigned long long( ULL);
const double eps( 1e-8);

/*Solution 1*/
class Solution
{
    //求A和B数组的第k大数
    int getMedian( int A [], int m , int B [], int n , int k)
    {
        if( m >n)
            return getMedian(B ,n , A , m , k); //默认A为短数组
        if( m == 0)
            return B [ k - 1 ];
        if( k == 1)
            return min( A [ 0 ], B [ 0 ]);
        int pa = min( k / 2 , m);
        int pb = k - pa;
        if( A [ pa - 1 ] < B [pb - 1 ])
        {
            return getMedian( A + pa , m - pa , B , n , k - pa);
        }
        else if( A [ pa - 1 ] > B [pb - 1 ])
        {
            return getMedian( A , m , B +pb , n -pb , k -pb);
        }
        else
        {
            return A [ pa - 1 ];
        }
        return 0;
    }
public :
    double work( int A [], int m , int B [], int n)
    {
        if(( m +n) % 2 == 0)
        {
            return ( getMedian( A , m ,B , n , ( m +n) / 2) + getMedian( A , m ,B , n , ( m +n) / 2 + 1)) / 2.;
        }
        else
        {
            return getMedian( A , m ,B , n , ( m +n) / 2 + 1);
        }
    }
    double findMedianSortedArrays( vector < int >& nums1 , vector < int >& nums2)
    {
        int A [ 10000 ],B [ 10000 ];
        int idx = 0;
        for( auto p: nums1)
        {
            A [ idx ++ ] =p;
        }
        idx = 0;
        for( auto p: nums2)
        {
           B [ idx ++ ] =p;
        }
        int m = nums1 . size();
        int n = nums2 . size();
        double ret = work( A , m ,B ,n);
        return ret;
    }
};

/*Solution 2*/
/*
class Solution
{
public:
   double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2)
   {
       auto it1=nums1.begin();
       auto it2=nums2.begin();
       vector<int> a;
       while(it1!=nums1.end() || it2!=nums2.end())
       {
           if(it1==nums1.end() && it2!=nums2.end())
           {
               a.push_back((*it2));
               it2++;
           }
           else if(it1!=nums1.end() && it2==nums2.end())
           {
               a.push_back(*it1);
               it1++;
           }
           else if(it1!=nums1.end() && it2!=nums2.end())
           {
               if((*it1)<(*it2))
               {
                   a.push_back(*it1);
                   it1++;
               }
               else
               {
                   a.push_back(*it2);
                   it2++;
               }
           }
       }
       int len=a.size();
       double ans=(len%2)?(double)a[len/2]:(double)(a[len/2-1]+a[len/2])/2.;
       return ans;
   }
};
*/

int main()
{
      int n , m , temp;
      while( cin >>n >> m)
      {
          vector < int > a ,b;
          for( int i = 0; i <n; ++ i)
          {
              cin >> temp;
              a . push_back( temp);
          }
          for( int i = 0; i < m; ++ i)
          {
              cin >> temp;
             b . push_back( temp);
          }
          Solution solution;
          double ans = solution . findMedianSortedArrays( a ,b);
          cout << "===========ans===========" << endl;
          cout << ans << endl;
      }
      return 0;
}

 

目录
相关文章
|
存储 缓存 算法
LeetCode刷题---Two Sum(一)
LeetCode刷题---Two Sum(一)
Leetcode 4. Median of Two Sorted Arrays
题目描述很简单,就是找到两个有序数组合并后的中位数,要求时间复杂度O(log (m+n))。 如果不要去时间复杂度,很容易就想到了归并排序,归并排序的时间复杂度是O(m+n),空间复杂度也是O(m+n),不满足题目要求,其实我开始也不知道怎么做,后来看了别人的博客才知道有个二分法求两个有序数组中第k大数的方法。
41 0
Leetcode Find Minimum in Rotated Sorted Array 题解
对一个有序数组翻转, 就是随机取前K个数,移动到数组的后面,然后让你找出最小的那个数,注意,K有可能是0,也就是没有翻转。
49 0
|
存储 C++ Python
LeetCode刷题---Add Two Numbers(一)
LeetCode刷题---Add Two Numbers(一)
|
存储 算法 安全
LeetCode - #2 Add Two Numbers
我们社区从本期开始会将顾毅(Netflix 增长黑客,《iOS 面试之道》作者,ACE 职业健身教练。)的 Swift 算法题题解整理为文字版以方便大家学习与阅读。 不积跬步,无以至千里;不积小流,无以成江海,Swift社区 伴你前行。
LeetCode - #2 Add Two Numbers
|
存储 算法 安全
LeetCode - #1 Two Sum
我们社区从本期开始会将顾毅(Netflix 增长黑客,《iOS 面试之道》作者,ACE 职业健身教练。)的 Swift 算法题题解整理为文字版以方便大家学习与阅读。 不积跬步,无以至千里;不积小流,无以成江海,Swift社区 伴你前行。
LeetCode - #1 Two Sum
LeetCode 167 Two Sum II - Input array is sorted(输入已排序数组,求其中两个数的和等于给定的数)
给定一个有序数组和一个目标值 找出数组中两个成员,两者之和为目标值,并顺序输出
88 0
|
算法 Python
LeetCode 108. 将有序数组转换为二叉搜索树 Convert Sorted Array to Binary Search Tree
LeetCode 108. 将有序数组转换为二叉搜索树 Convert Sorted Array to Binary Search Tree
|
算法 测试技术
LeetCode 88. 合并两个有序数组 Merge Sorted Array
LeetCode 88. 合并两个有序数组 Merge Sorted Array
|
3月前
|
Unix Shell Linux
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
本文提供了几个Linux shell脚本编程问题的解决方案,包括转置文件内容、统计词频、验证有效电话号码和提取文件的第十行,每个问题都给出了至少一种实现方法。
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行