สมมติว่าเรามีรายการที่เรียงลำดับสองรายการ เราต้องหาค่ามัธยฐานของสองรายการนี้ ดังนั้นหากอาร์เรย์เป็นเหมือน [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
- ถ้า totalLength mod 2 เหมือนกับ 0 แล้ว:
- มิฉะนั้นเมื่อ 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