leetcode C++ 4. 寻找两个有序数组的中位数 给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。 请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log
生活随笔
收集整理的這篇文章主要介紹了
leetcode C++ 4. 寻找两个有序数组的中位数 给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。 请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
一、概述:
? ? ?這道題思路其實蠻簡單的,就是邊界太難考慮了,所以一直沒能通過,寫了好長的代碼考慮邊界,然后看到一個大神寫的代碼:
https://leetcode-cn.com/problems/median-of-two-sorted-arrays/solution/c-er-fen-cha-zhao-by-da-li-wang/?。簡直是嘆為天人,敬佩不已,發現自己走的路還很長。
二、思路:
? ? 由于是兩個排序的數組,只要兩個數組的間距為保持為總長的一半,二分遍歷較短的數組,然后通過一半思路去求另一個數組的位置,比較數組的左邊是否小于另一數組當前節點,如果是,找到中位數的位置。
? ?特別需要注意的是,在較長的數組上一定存在一個中位數。
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {if (nums1.size() > nums2.size())return findMedianSortedArrays(nums2, nums1);int M = nums1.size();int N = nums2.size();int l = 0;int r = M;while (l <= r) {int mid = (r - l) / 2 + l;int rightPos = (N + M) / 2 - mid;int max1 = (mid - 1 >= 0) ? nums1[mid - 1] : INT_MIN;int min1 = (mid < M) ? nums1[mid] : INT_MAX;int max2 = (rightPos - 1 >= 0) ? nums2[rightPos - 1] : INT_MIN;int min2 = (rightPos < N) ? nums2[rightPos] : INT_MAX;if (max1 > min2) { //如果num1[mid]左邊的數大于右邊指向的數,中位數左移r = mid - 1;}else if (max2 > min1) { //如果num1[mid]右邊的數大于左邊指向的數,中位數右移l = mid + 1;}else {l = r = mid;break;}}int u = l; //指向nums1對應的中位數位置int v = (M + N) / 2 - l; //指向中位數右邊對應中位數的位置int low = max(u - 1 >= 0 ? nums1[u - 1] : INT_MIN,v - 1 >= 0 ? nums2[v - 1] : INT_MIN);int high = min(u < M ? nums1[u] : INT_MAX,v < N ? nums2[v] : INT_MAX);if ((N + M) % 2 == 1)return high;return float(low + high) / 2; }?
總結
以上是生活随笔為你收集整理的leetcode C++ 4. 寻找两个有序数组的中位数 给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。 请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: C++ leetcode 给你一个数组
- 下一篇: leetcode C++ 2. 两数相加