[LeetCode] 88. Merge Sorted Array

原题链接: https://leetcode.com/problems/merge-sorted-array/

1. 题目介绍

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 n respectively.
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2.

给定两个有序整数数组,把数组 2 中的元素都放入数组 1 中,同时还要保证数组 1 是有序的。数组 1 中实际存放了 m 个元素,数组 2 实际存放了 n 个元素。题目保证数组 1 的长度是大于等于 m + n 的,也就是,数组1装得下所有的元素。

Example:

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

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

2. 解题思路

倒着存放数组1。
数组1 中第m个元素之后的空间都是闲置的,改变它们的值不会影响题目给出的元素。所以先从后面开始排序比较好。
从数组1的第 m+n 个元素向前遍历,依次放入数组1 和数组 2 中较大的数。然后就可以成功完成排序了。
某一个数组可能在早期就被用完了,所以为了防止指针异常,循环中要加入判断该数组是否已经被用完的判断。

实现代码

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
       
        int length = m + n;
        int curA = m-1;
        int curB = n-1;

        for (int i = length - 1 ; i >= 0 ; i-- ) {

        	if (curA <0) {
        		nums1[i] = nums2[curB];
        		curB --;
        	}
        	else if (curB <0) {
        		nums1[i] = nums1[curA];
        		curA --;
        	}
        	else if ( nums1[curA] > nums2[curB]  ) {
        		nums1[i] = nums1[curA];
        		curA --;
        	}else{
        		nums1[i] = nums2[curB];
        		curB --;
        	}
        }
    }
}

还有一个更简洁的写法,这个思路来自 https://www.cnblogs.com/grandyang/p/4059650.html ,有兴趣的读者可以直接去看原文。
这个做法是还是从数组1 后部闲置的空间向前遍历,只是循环结束的条件成了判断数组2 是否为空。

实现代码

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
       
        int length = m + n -1;
        int curA = m-1;
        int curB = n-1;
        while ( curB >= 0 ) {
        	if (curA >= 0 && nums1[curA] > nums2[curB]) {
        		nums1[length] = nums1[curA];
        		length--;
        		curA--;
        	}else{
        		nums1[length] = nums2[curB];
        		length--;
        		curB--;
        	}
        }
    }
}

3. 参考资料

https://www.cnblogs.com/grandyang/p/4059650.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值