zoukankan      html  css  js  c++  java
  • 每天一道算法题目(18)——取等长有序数组的上中位数和不等长有序数组的第k小的数

    1.取上中位数

    题目

              给定两个有序数组arr1和arr2,两个数组长度都为N,求两个数组中所有数的上中位数。要求:时间复杂度O(logN)。
         例如:
             arr1 = {1,2,3,4};
             arr2 = {3,4,5,6};
             一共8个数则上中位数是第4个数,所以返回3。

             arr1 = {0,1,2};
             arr2 = {3,4,5};
             一共6个数则上中位数是第3个数,所以返回2。

    思路

         (1)假设两个数组长度为偶数,取中间节点index=1
                1  2  3  4
                1‘ 2’ 3‘ 4’
             若2 == 2‘ ,则直接返回;
             若2 > 2', 说明 2 至少排第4, 所以3,4可以排除,1’ 2‘(2’ 最多排第3)可以排除,所以对剩下的1 2和3‘ 4’ 递归
             若2 < 2‘,同理递归1’ 2‘和3、4.

     

          (2)假设两个数组长度为奇数,取中间节点index=2.
              1  2  3  4  5
              1‘ 2’ 3‘ 4’ 5’
             若3 == 3‘ ,则直接返回;
             若3 > 3', 说明 3 至少排第6, 所以4,5可以排除,1’ 2‘(2’ 最多排第4)可以排除,所以对剩下的1 2 3和 3‘ 4’  5‘递归,其实3也能排除,但是为了保证两个数组的长度一样,保留3
             若3 < 3‘,同理递归1’ 2‘ 3’和3、4 、5.

    代码

    int getUpMedian(const vector<int> & arr1, int start1, int end1,const vector<int> & arr2, int start2, int end2) {
      
                if(start1 == end1)
                    return min(arr1[start1], arr2[start2]);
    
                int size = end1-start1+1;
                int halfSize;
      
                if(size&1 == 1)//奇
                    halfSize = (size + 1)/2;
                else//偶
                    halfSize = size/2;
            
      
                if(arr1[start1+halfSize - 1] == arr2[start2 + halfSize - 1])
                    return arr1[start1 + halfSize - 1];
                else if(arr1[start1 + halfSize - 1] > arr2[start2 + halfSize - 1])
                    return getUpMedian(arr1,start1,start1+halfSize-1,arr2,end2-(halfSize-1),end2);
                else
                    return getUpMedian(arr1,end1-(halfSize-1),end1,arr2,start2,start2+halfSize -1);
    }
    
    int getUpMedian(vector<int> arr1, vector<int> arr2) {//主函数
      
            if(arr1.size() != arr2.size())
                return -1;
            if(arr1.size() == 0)
                return -1;
            return getUpMedian1(arr1,0,arr1.size()-1,arr2,0,arr1.size() -1 );
    }


    2.取不等长有序数组的最小K个数

    题目

           给定两个有序数组arr1和arr2,在给定一个整数k,返回两个数组的所有数中第K小的数。要求:如果arr1的长度为N,arr2的长度为M。要求时间复杂度达到O(log(min{M,N}))
          例如:arr1 = {1,2,3,4,5};arr2 = {3,4,5};K = 1; 因为1为所有数中最小的,所以返回1;
                     arr1 = {1,2,3};arr2 = {3,4,5,6};K = 4;因为3为所有数中第4小的数,所以返回3;

    思路

         分三种情况考虑,假设arr1的长度N小于等于arr2的长度M

         (1)若K<=N 

                 结果为 arr1[0---K-1]和arr2[0---K-1]取中位数

         (2)若 N<K<=M。先将arr2分为[0--K-N-1],[K-N-----K-1],[K,M-1]三部分

                 当arr2[K-N-1]>=arr1[N-1],结果为arr2[K-N-1]

                 否则,arr2[0--K-N-1]部分的所有数据一定至多在K-1位置(极端情况下,arr1[N]占据K位置)。而arr2[K,M-1]区间段的元素至少在K+1位置。因此结果对长度为N的两段区间段arr1[0,N-1]和arr2[K-N,K-1]取中位数即可。

         (3)若K>M>N。先将划分arr1[0,K-M-1][K-M,N-1]两段区间,arr2[0,K-N-1][K-N,M-1]

                当arr2[K-N-1]>=arr1[N-1],结果为arr2[K-N-1]

                当arr1[K-M-1]>=arr2[M-1],结果为arr1[K-M-1]

                两者不满足的话。则如前所叙述,arr1[0,K-M-1]和arr2[0,K-N-1]的元素都在位置K之前,至多为K-1。因此结果为长度为M+N-K的两段区间arr1[K-M,N-1]和arr2[K-N,M-1]的中位数。


         时间复杂度最大为情况(2),即为所要求。


    代码

    int findKthNum(vector<int> arr1, vector<int> arr2, int kth)
            {
                int size1 = arr1.size();
                int size2 = arr2.size();
     
                if(size1 > size2)
                    return findKthNum(arr2, arr1, kth);//ensur
    
               if(kth <= size1)
                    return getUpMedian(arr1, 0, kth - 1, arr2, 0, kth - 1);//前k个,取中位数即可
               else if (kth <= size2 && kth > size1)
               {
                    if(arr2[kth - size1 - 1] >= arr1[size1 - 1] )
                        return arr2[kth - size1 - 1];
    				else
                       return getUpMedian(arr1, 0, size1-1, arr2, kth-size1, kth-1);
               }
               else 
               {
                    if(arr2[kth-size1-1] >= arr1[size1-1])
                        return arr2[kth-size1-1];
                    if(arr1[kth-size2-1] >= arr2[size2-1])
                        return arr1[kth-size2-1];
                   return getUpMedian(arr1, kth-size2, size1-1, arr2, kth-size1, size2-1);
               }
     
    }






    参考
       1.[nowCoder] 两个长度相同有序数组的中位数

                             

                   

  • 相关阅读:
    拉普拉斯------拉普拉斯算子
    拉普拉斯------拉普拉斯变换
    傅里叶------傅里叶变换
    傅里叶------傅里叶级数
    深度学习笔记------卷积训练
    深度学习笔记------卷积神经网络
    计算机图形学笔记------直线的扫描转换
    深度学习笔记------神经网络
    深度学习笔记------softmax回归
    深度学习笔记------线性回归
  • 原文地址:https://www.cnblogs.com/engineerLF/p/5393019.html
Copyright © 2011-2022 走看看