合并两个有序数组【leetcode】

短命女 2023-10-14 23:16 131阅读 0赞

### 1. 题目： ###

给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2，另有两个整数 m 和 n ，分别表示 nums1 和 nums2 中的元素数目。

请你 合并 nums2 到 nums1 中，使合并后的数组同样按 非递减顺序 排列。

注意：最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。为了应对这种情况，nums1 的初始长度为 m + n，其中前 m 个元素表示应合并的元素，后 n 个元素为 0 ，应忽略。nums2 的长度为 n 。

### 2. 思路： ###

每次比较两个数组中最大的元素即可。

① 首先分别拿两个数组中最大的元素出来作比较，较大的元素放在 num1 数组的 m+n-1 位置处，则当前元素的原位置索引（m 或 n）减减，目的指向下一个待比较的元素，而我们的 num1 空位置索引也要减减，目的指向下一个空位置，这里的空位置索引我们就用 m+n-1 来表示，当 m 或 n 减减的同时，该空位置索引也会随着减减；

② 这个比较过程是要重复进行的，什么时候停止？当比较的位置已经到达了最左边，即索引为 0 的时候，我们就不可能再去比较了，因为这里的 m-1 和 n-1 才是两个数组各自的索引，则循环条件就是 m > 0 && n > 0，只要有一个数组遍历到了尽头，就退出循环；

③ 循环不满足条件退出之后，如果 n<0，但 m>0 或者 n 和 m 都小于 0，就代表 num2 中的元素已经全部插入到 num1 中了，这个时候就算合并完成；

④ 如果 m<0，但 n>0，这种情况就是 num1 中的元素已经全比较完了，只剩下 num2 的元素等待插入，所以这个时候我们还需要一个 while 循环，用于将 num2 中的元素依次插入到 num1 中，空位置索引还是 m+n-1，每次 n 减减，直到 n 等于0。

### 3. 代码实现： ###

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