合并两个有序数组

题目:

给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。

说明:

初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。

示例:

示例:

输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3

输出: [1,2,2,3,5,6]

思路:

①:丑陋的代码 时间复杂度 O(n^2)
组合以后再排序

System.arraycopy(nums2, 0, nums1, m, n);
Arrays.sort(nums1);

②美丽的代码 时间复杂度O(n+m)
双指针从前往后,最直接的算法实现是将指针p1 置为 nums1的开头, p2为 nums2的开头,在每一步将最小值放入输出数组中。由于 nums1 是用于输出的数组,需要将nums1中的前m个元素放在其他地方,也就需要 O(m) 的空间复杂度。

③更美丽的代码 时间复杂度 : O(n + m) 空间复杂度 : O(1)。
双指针 / 从后往前


双指针

代码:

//①:丑陋的代码 时间复杂度 O(n^2)
class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        
        int temp = 0;
        int index = m; //index = m = 3
        for (int i : nums2) {
            nums1[index] = i;//nums[3] = 2
            index++;//index = 4
        }//nums1 = 1,2,3,2,5,6
        
        for (int j = 0; j < nums1.length - 1; j++){
            for (int i = 0; i < nums1.length - 1 - j; i++) {
                if (nums1[i+1] < nums1[i]) {
                    temp = nums1[i];
                    nums1[i] = nums1[i+1];
                    nums1[i+1] = temp;
                }
            }
        }
        
    }
}

//②美丽的代码 时间复杂度O(n+m) 空间复杂度 : O(m)
class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        int newNums[] = new int[m];
        System.arraycopy(nums1, 0, newNums, 0, m);
        int p = 0;
        int p1 = 0;
        int p2 = 0;
        while (p1 < m && p2 < n) {
            if (newNums[p1] > nums2[p2]) {
                nums1[p] = nums2[p2];
                p2++;
            } else {
                nums1[p] = newNums[p1];
                p1++;
            }
            p++;
        }
        if (p1 < m) {
            System.arraycopy(newNums, p1, nums1, p, m-p1);
        }
        if (p2 < n) {
            System.arraycopy(nums2, p2, nums1, p, n-p2);
        }
        

    }
}

//更美丽的代码 时间复杂度 : O(n + m) 空间复杂度 : O(1)。
class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        
        int p = m + n - 1;
        int p1 = m - 1;
        int p2 = n - 1;
        while (p1 >= 0 && p2 >= 0) {
            if (nums2[p2] > nums1[p1]) {
                nums1[p] = nums2[p2];
                p2--;
            }else {
                nums1[p] = nums1[p1];
                p1--;
            }
            p--;
        }
        System.arraycopy(nums2, 0, nums1, 0, p2+1);
        
        
    }
}

你可能感兴趣的:(合并两个有序数组)