原题链接: 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--;
}
}
}
}