面试题：合并2个有序数组（leetcode88）

Posted 2020-08-10 昕友软件开发

给定两个有序整数数组 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(m+n)

package javaDemo;

public class ArrDemo {
    
    public static void main(String[] args) {
        int[] nums1 =  new int[] {1,2,3};
        int[] nums2 =  new int[] {2,5,6};
        
        
        for (int num : merge(nums1,3,nums2,3)) {
            System.out.println(num);
        }
            

    }
    
    public static int[] merge(int[] nums1, int m, int[] nums2, int n) {
        int[] k =new int[m+n];
        int i=0,j=0,kindex=0;
        while (i<m && j<n && kindex<m+n){
            if (nums1[i] <= nums2[j]){
                k[kindex++] = nums1[i];
                i++;                    
            }
            else{
                k[kindex++] = nums2[j];
                j++;
            }
            
            //说明nums1已经结束遍历，把nums2的剩余全给k
            if (i==m) {
                while(j<n) {
                    k[kindex++] = nums2[j++];  
                }
            }

            if (j==n) {
                while(i<m) {
                    k[kindex++] = nums1[i++];  
                }
            }
        }
        
        nums1 = k;
        return  nums1;
        
    }

}

还有一个更高效的:

public void merge(int[] nums1, int m, int[] nums2, int n) {
        int p = m-- + n-- - 1;
        while (m >= 0 && n >= 0) {
            nums1[p--] = nums1[m] > nums2[n] ? nums1[m--] : nums2[n--];
        }
        
        while (n >= 0) {
            nums1[p--] = nums2[n--];
        }
        
    }