[88] Merge Sorted Array

来源:互联网 发布:手机计算器软件 编辑:程序博客网 时间:2024/04/30 10:10

1. 题目描述

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 nums1 and nums2 are m and n respectively.

给定两个有序数组,将数组2合并到数组1中。其中数组1原有的长度足够(不必使用额外空间),且给定了数组1和数组2中原有元素的个数分别为m和n。

2. 解题思路

受了程序员面试金典111页1.4题的启发(我也不知道为啥突然想到这个题),在不利用额外空间的情况下,可以选择从后往前做。从后往前做的方案就是,使用三个指针,一个s指向数组1的当前结尾,一个t指向数组2的当前结尾,一个i指向数组1在合并1和2之后的结尾,那么我们就可以知道,这个第i个值的取值方式就是从数组1中的第s个值和数组2第t个值中选择一个大的,之后将那个指针前移,继续进行比较,直到其中一个数组的指针到达头部则结束。说到这个到达头部,还分几种情况,一种是s先到达头部,那么如果此时i没有到达头部的话,证明了,数组2中剩余的元素都是排在0-i的元素,进行一次拷贝即可。如果t先到达头部,那么证明,已经将数组2中所有元素都插入到了数组1中,所以直接结束即可。上面的图描述了1先做完的情况,下面的图描述了2先做完的情况。
1先做完
2先做完

3. Code

public class Solution {    public void merge(int[] nums1, int m, int[] nums2, int n) {        int s = m-1;  // 指向nums1的最后一个        int t = n-1;  // 指向nums2的最后一个        // 从新数组的最后一个元素开始设置        for(int i = m+n-1; i >= 0; --i)        {            // 当nums1数组已经做完时,将nums2剩余的部分拷贝到数组前端            if(s < 0)            {                System.arraycopy(nums2, 0, nums1, 0, t+1); // 剩余的num2接到前面的部分                break;            }            // 当nums2已经做完时,证明已经插入完毕            else if(t < 0)            {                break;            }            else{            // 当前的值为数组1和2中较大的那个值,且较大值得指针前移一个                nums1[i] = nums2[t] >= nums1[s] ? nums2[t--] : nums1[s--];            }        }    }}
0 0