Trung vị của hai mảng được sắp xếp trong C ++

Giả sử chúng ta có hai mảng; các mảng này được sắp xếp. Vì vậy ta phải tìm trung vị của hai mảng này. Vì vậy, nếu các mảng giống như [1,5,8] và [2,3,6,9], thì câu trả lời sẽ là 5.

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -

• Xác định một hàm findMedianSortedArrays, điều này sẽ lấy mảng nums1 và nums2

• nếu kích thước của nums1> kích thước của nums2, thì

  • Gọi hàm trả về findMedianSortedArrays (nums2, nums1)

• x:=kích thước của nums1, y:=kích thước của nums2

• thấp:=0, cao:=x

• TotalLength:=x + y

• trong khi thấp <=cao, làm -

  • partitionX:=low + (high - low) / 2

  • partitionY:=(totalLength + 1) / 2 - partitionX

  • maxLeftX =-inf khi partitionX bằng 0, nếu không thì nums1 [partitionX-1]

  • maxRightX =inf khi partitionX là x, ngược lại là nums1 [partitionX]

  • maxLeftY =-inf khi phân vùngY bằng 0, nếu không thì nums2 [phân vùngY-1]

  • maxRightY =inf khi phân vùngY là y, ngược lại nums2 [phân vùngY]

  • nếu maxLeftX <=minRightY và maxLeftY <=minRightX, thì

    • nếu totalLength mod 2 giống 0, thì

      • return (tối đa maxLeftX và maxLeftY) + tối thiểu minRightX và minRightY) / 2

    • Nếu không

      • trả về tối đa maxLeftX và maxLeftY

  • Ngược lại khi maxLeftX> minRightY thì -

    • cao:=partitionX - 1

  • Nếu không thì thấp:=partitionX + 1

• trả về 0

Ví dụ (C ++)

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -

#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
   double findMedianSortedArrays(vector& nums1, vector<int>& nums2) {
      if(nums1.size()>nums2.size())
         return findMedianSortedArrays(nums2,nums1);
      int x = nums1.size();
      int y = nums2.size();
      int low = 0;
      int high = x;
      int totalLength = x+y;
      while(low<=high){
         int partitionX = low + (high - low)/2;
         int partitionY = (totalLength + 1)/2 - partitionX;
         int maxLeftX = (partitionX ==0?INT_MIN:nums1[partitionX1] );
         int minRightX = (partitionX == x?INT_MAX :
         nums1[partitionX]);
         int maxLeftY = (partitionY ==0?INT_MIN:nums2[partitionY1] );
         int minRightY = (partitionY == y?INT_MAX : nums2[partitionY]);
         if(maxLeftX<=minRightY && maxLeftY <= minRightX){
            if(totalLength% 2 == 0){
               return ((double)max(maxLeftX,maxLeftY) + (double)min(minRightX,minRightY))/2;
            } else {
               return max(maxLeftX, maxLeftY);
            }
         }
         else if(maxLeftX>minRightY)
            high = partitionX-1;
         else low = partitionX+1;
      }
      return 0;
   }
};
main(){
   Solution ob;
   vector<int> v1 = {1,5,8}, v2 = {2,3,6,9};
   cout << (ob.findMedianSortedArrays(v1, v2));
}

Đầu vào

[1,5,8]
[2,3,6,9]

Đầu ra

5