LeetCode刷题记录

88. 合并两个有序数组

2019-06-15  本文已影响3人  046ef6b0df68

文|Seraph

01 | 问题

给定两个有序整数数组 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]

02 |解题

初解:

两个数组大小都知道,所以合并后的大小也是确定的。从后往前合并数组,减少没必要的挪位操作。
当然也可以从前往后,只是需要额外的O(n+m)空间。

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int size=m+n;
        int k=m-1;
        int j=n-1;
        for(int i=size-1; i>=0; i--)
        {
            if(k<0 || j<0)
                break;
            if(nums1[k]<nums2[j]) 
            {
                nums1[i]=nums2[j];
                j--;
            }
            else
            {
                nums1[i]=nums1[k];
                k--;
            }
        }
        for(int i=0; i<=j; i++)//如果nums2还有数,需要移动到nums1中
            nums1[i]=nums2[i];
    }
};

终解:

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

03 | 积累知识点

  1. 三目条件运算符经常可以和判断语句有等同的效果。虽然不推荐使用,因为复杂的判断逻辑交错在一起,往往会不清晰。但是简单的判断逻辑的情况,使用还是挺好的。

官网链接 Leetcode No.88

上一篇下一篇

猜你喜欢

热点阅读