技術標籤：LeetCode演算法leetcodec++歸併排序

上一篇部落格：LeetCode 605. 種花問題（簡單模擬）——每日一題

 寫在前面：大家好！我是AC-fun,我的暱稱來自兩個單詞Accepted和fun。我是一個熱愛ACM的蒟蒻。最近萌生了刷LeetCode的想法，所以我打算從LeetCode簡單的題目開始做起，攻陷LeetCode。如果部落格中有不足或者的錯誤的地方歡迎在評論區或者私信我指正，感謝大家的不吝賜教。我的唯一部落格更新地址是：https://ac-fun.blog.csdn.net/。非常感謝大家的支援。一起加油，衝鴨！
 用知識改變命運，用知識成就未來！加油 (ง •̀o•́)ง (ง •̀o•́)ง

原題連結：LeetCode 4. 尋找兩個正序陣列的中位數

文章目錄

• 題目資訊
• • 題目描述
  • 示例 1
  • 示例 2
  • 示例 3
  • 示例 4
  • 示例 5
  • 提示：
• 題解
• • 解題思路
  • 解題程式碼

題目資訊

題目描述

 給定兩個大小為 m 和 n 的正序（從小到大）陣列 nums1 和 nums2。請你找出並返回這兩個正序陣列的中位數。

進階：你能設計一個時間複雜度為 O(log (m+n)) 的演算法解決此問題嗎？

示例 1

輸入：nums1 = [1,3], nums2 = [2]
輸出：2.00000
解釋：合併陣列 = [1,2,3] ，中位數 2

示例 2

輸入：nums1 = [1,2], nums2 = [3,4]

輸出：2.50000
解釋：合併陣列 = [1,2,3,4] ，中位數 (2 + 3) / 2 = 2.5

示例 3

輸入：nums1 = [0,0], nums2 = [0,0]
輸出：0.00000

示例 4

輸入：nums1 = [], nums2 = [1]
輸出：1.00000

示例 5

輸入：nums1 = [2], nums2 = []
輸出：2.00000

提示：

• nums1.length == m
• nums2.length == n
• 0 <= m <= 1000
• 0 <= n <= 1000
• 1 <= m + n <= 2000
• -106 <= nums1[i], nums2[i] <= 106

題解

解題思路

 首先使用 歸併排序 將兩個陣列從小到大進行排序，並放到另一個 vector 陣列 ans 中，然後再找到並計算出新陣列的中間值。如果結果陣列 ans 的長度為奇數那麼中間值只有一個，此時直接返回 double 型別的中間值即可。如果 ans 的長度為偶數，那麼中間值有兩個，需要將兩個中間值相加除以 2 之後再返回。有關歸併排序的詳細解釋請看 經典排序演算法之歸併排序詳解。

解題程式碼

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ans;
        int len1 = nums1.size(), len2 = nums2.size();

        // 進行歸併排序
        int i = 0, j = 0;
        while (i < len1 && j < len2) {
            if (nums1[i] <= nums2[j]) ans.push_back(nums1[i++]);
            else ans.push_back(nums2[j++]);
        }
        while (i < len1) ans.push_back(nums1[i++]);
        while (j < len2) ans.push_back(nums2[j++]);

        // 計算並返回結果
        int len3 = ans.size();
        double res = 0;
        if (len3 % 2 != 0) res = (double)ans[len3 / 2];
        else res = (double)(ans[len3 / 2] + ans[(len3 / 2) - 1]) / 2;
        return res;
    }
};

未完待續，持續更新中……