Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> C++

โปรแกรมหาค่ามัธยฐานของรายการที่เรียงลำดับสองรายการใน C++


สมมติว่าเรามีรายการที่เรียงลำดับสองรายการ เราต้องหาค่ามัธยฐานของสองรายการนี้ ดังนั้นหากอาร์เรย์เป็นเหมือน [1,5,8] และ [2,3,6,9] คำตอบจะเป็น 5

เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้:

  • กำหนดฟังก์ชัน Solve() ซึ่งจะใช้อาร์เรย์ nums1 อาร์เรย์ nums2
  • ถ้าขนาดของ nums1> ขนาดของ nums2 แล้ว:
    • ผลตอบแทนการแก้(nums2, nums1)
  • x :=ขนาดของ nums1, y :=ขนาดของ nums2
  • ต่ำ :=0, สูง :=x
  • ความยาวทั้งหมด :=x + y
  • ในขณะที่ต่ำ <=สูง ทำ:
    • partitionX :=ต่ำ + (สูง - ต่ำ)
    • partitionY :=(totalLength + 1) / 2 - partitionX
    • maxLeftX :=(หาก partitionX เหมือนกับ 0 แล้ว -inf มิฉะนั้น nums1[partitionX - 1])
    • minRightX :=(หาก partitionX เหมือนกับ x แล้ว inf มิฉะนั้น nums1[partitionX])
    • maxLeftY :=(หาก partitionY เหมือนกับ 0 แล้ว -inf มิฉะนั้น nums2[partitionY - 1])
    • minRightY :=(หาก partitionY เหมือนกับ y แล้ว inf มิฉะนั้น nums2[partitionY])
    • ถ้า maxLeftX <=minRightY และ maxLeftY <=minRightX แล้ว:
      • ถ้า totalLength mod 2 เหมือนกับ 0 แล้ว:
        • ผลตอบแทน ((สูงสุด maxLeftX และ maxLeftY) + (ขั้นต่ำ minRightX และ minRightY))/ 2
      • มิฉะนั้น
        • ส่งคืนสูงสุดของ maxLeftX และ maxLeftY
    • มิฉะนั้นเมื่อ maxLeftX> minRightY แล้ว:
      • สูง :=partitionX - 1
    • มิฉะนั้น
  • คืน 0

ให้เราดูการใช้งานต่อไปนี้เพื่อความเข้าใจที่ดีขึ้น:

ตัวอย่าง

#include
using namespace std;

class Solution {
   public:
   double solve(vector& nums1, vector& nums2) {
      if(nums1.size()>nums2.size())
         return solve(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[partitionX-1] );
         int minRightX = (partitionX == x?INT_MAX : nums1[partitionX]);

         int maxLeftY = (partitionY ==0?INT_MIN:nums2[partitionY-1] );
         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 v1 = {1,5,8}, v2 = {2,3,6,9};
   cout << (ob.solve(v1, v2));
}

อินพุต

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

ผลลัพธ์

5