1. 程式人生 > >LeetCode算法系列:88.Merge Sorted Array

LeetCode算法系列:88.Merge Sorted Array

題目描述:

Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.

Note:

  • The number of elements initialized in nums1 and nums2 are m and nrespectively.
  • You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2
    .

Example:

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

Output: [1,2,2,3,5,6]

演算法實現:

這個問題比較簡單,題目中也沒有對空間和時間的要求,最直接的可以直接新建一個數組來進行排序也行。

我們可以利用nums1後面的空間,做到只利用常數空間完成要求

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int i = m - 1, j = n - 1, k = m + n - 1;
        while(i >= 0 && j >= 0){
            if(nums1[i] >= nums2[j]){
                nums1[k] = nums1[i];
                k --;
                i --;
            }
            else{
                nums1[k] = nums2[j];
                k --;
                j --;
            }
        }
        if(i = -1){
            for(int t = 0; t <= j; t ++)
                nums1[t] = nums2[t];
        }
    }
};