leetcode学习笔记：Merge Sorted Array

一.题目描述

二.解题技巧

这道题不存在复杂的分析过程和边界条件。如果单纯得考虑从小到大地将两个数组进行合并的话，每次在num1中插入一个数的话，需要将后面的元素都向后移动一位，这样，整个处理过程的时间复杂度为O(m*n)。

由于两个数组的元素的个数是知道的，同时，合并后的数组也是递增排序的，也就是说，排序之后的数组的最大值是放在最后面的，因此，我们可以从后往前遍历，也就是将最大值放在第一个数组的m+n-1位置，然后将次最大值放在m+n-2位置，依次类推，这样在将元素放置到合适位置的时候，就不需要移动元素，这个方法的时间复杂度为O(m+n)。

三.示例代码

// 时间复杂度O(m+n)，空间复杂度O(1)
class Solution {
public:
    void merge(int A[], int m, int B[], int n) {
        int ia = m - 1, ib = n - 1, icur = m + n - 1;
        while (ia >= 0 && ib >= 0) {
            A[icur--] = A[ia] >= B[ib] ? A[ia--] : B[ib--];
        }
        while (ib >= 0) {
            A[icur--] = B[ib--];
        }
    }
};

// 使用STL
#include <iostream>  
#include <vector>  

using std::vector;

class Solution
{
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n)
    {
        int ResultIndex = m + n - 1;
        m--;
        n--;
        while (m >= 0 || n >= 0)
        {
            if (m < 0)
            {
                nums1[ResultIndex--] = nums2[n--];
                continue;
            }

            if (n < 0)
            {
                nums1[ResultIndex--] = nums1[m--];
                continue;
            }

            if (m >= 0 && n >= 0)
            {
                if (nums1[m] > nums2[n])
                {
                    nums1[ResultIndex--] = nums1[m--];
                    continue;
                }
                else
                {
                    nums1[ResultIndex--] = nums2[n--];
                    continue;
                }
            }
        }
    }
};