1. 程式人生 > >LeetCode--陣列類演算法--合併兩個有序陣列

LeetCode--陣列類演算法--合併兩個有序陣列

給定兩個有序整數陣列 nums1 和 nums2,將 nums2 合併到 nums1 使得 num1 成為一個有序陣列。

說明:

  • 初始化 nums1 和 nums2 的元素數量分別為 m 和 n
  • 你可以假設 nums1 有足夠的空間(空間大小大於或等於 m + n)來儲存 nums2 中的元素。

示例:

輸入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6],       n = 3

輸出: [1,2,2,3,5,6]

思路:把nums2接到nums1上,隨後排序。

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
       int a=m;
        for(int i=0;i<n;i++)
       {
           nums1[a++]=nums2[i];
       }
        sort(nums1.begin(),nums1.end(),greater<int>() );//從大到小排序
        nums1.resize(m+n);//限制nums1的大小,多出的部分被刪掉
        sort(nums1.begin(),nums1.end());//實際上翻轉就可以。
    }
};