文章

每日LeetCode 24

2021.3.12 ・ 共 236 字,您可能需要 1 分钟阅读

Tags: LeetCode

给你两个有序整数数组 nums1nums2,请你将 nums2 合并到 nums1 中*,*使 nums1 成为一个有序数组。

初始化 nums1nums2 的元素数量分别为 mn 。你可以假设 nums1 的空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素。

示例 1:

输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]
class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        if (n == 0) {
            return;
        }
        int[] nums1Copy = new int[m + n];
        System.arraycopy(nums1, 0, nums1Copy, 0, m + n);
        int a = 0;
        int b = 0;
        for (int i = 0; i < m + n; i++) {
            if (a > m - 1) {
                nums1[i] = nums2[b];
                b++;
                continue;
            }
            if (b > n - 1) {
                nums1[i] = nums1Copy[a];
                a++;
                continue;
            }

            if (nums1Copy[a] <= nums2[b]) {
                nums1[i] = nums1Copy[a];
                a++;
            } else {
                nums1[i] = nums2[b];
                b++;
            }

        }
    }
}