Leetcode.合并两个有序数组(Java实现)

/**
 * 合并两个有序数组
 * 给定两个有序整数数组 nums1 和 nums2，将 nums2 合并到 nums1 中，使得 num1 成为一个有序数组。
 * <p>
 * 说明:
 * <p>
 * 初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
 * 你可以假设 nums1 有足够的空间（空间大小大于或等于 m + n）来保存 nums2 中的元素。
 * 示例:
 * <p>
 * 输入:
 * nums1 = [1,2,3,0,0,0], m = 3
 * nums2 = [2,5,6],       n = 3
 * <p>
 * 输出: [1,2,2,3,5,6]
 * <p>
 * 解题思路：定义一个新数组，定义三个指针分别指向三个数组的第一个元素，比较两个数组的大小，将小的元素放入新数组，当一个数组放入结束后就将另一个全部放入新数组中，最后将新数组的元素赋值给nums1数组
 */

public class MergeArray {
    public static void main(String[] args) {
        int[] nums1 = {1, 2, 3,0,0,0};
        int[] nums2 = {2, 5, 6};
        merge(nums1, 3, nums2, 3);
    }

    public static void merge(int[] nums1, int m, int[] nums2, int n) {
        int[] newArray = new int[m + n];//定义一个新的数组，长度为两个原数组长度之和
        int i = 0, j = 0, count = 0;//定义三个指针，分别指向三个数组的第一个元素
        //当指针位置小于数组长度时，比较两个数组的元素的大小，小的放入新数组
        while (i < m && j < n) {
            if (nums1[i] < nums2[j]) {
                newArray[count++] = nums1[i++];
            } else {
                newArray[count++] = nums2[j++];
            }
        }

        if (i >= m) {//当第一个数组遍历完成，直接将第二个数组元素放入新数组
            while (j < n) {
                newArray[count++] = nums2[j++];
            }
        }
        if (j <= n) {//当第一个数组遍历完成，直接将第二个数组元素放入新数组
            while (i < m) {
                newArray[count++] = nums1[i++];
            }
        }
        for (int k=0;k<newArray.length;k++){//将新数组的元素依次赋值给nums1
            nums1[k]=newArray[k];
        }

    }
}