Fork me on GitHub

【LeetCode】88. Merge Sorted Array

题目:

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

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

提示:

此题在题干中已经提示了nums1的大小是nums1与nums2的有效数字大小之和,因此我们可以从后往前(从大到小)排序,这样就仅涉及到赋值操作而不需要swap。

代码:

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

 在论坛中发现了一种非常简洁的写法:

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

两者的思路是一样的。

posted @ 2015-08-31 15:57  __Neo  阅读(168)  评论(0编辑  收藏  举报