码迷,mamicode.com
首页 > 编程语言 > 详细

面试题:合并2个有序数组(leetcode88)

时间:2019-04-12 13:47:56      阅读:172      评论:0      收藏:0      [点我收藏+]

标签:nbsp   div   pre   空间   col   有序   style   leetcode   span   

给定两个有序整数数组 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--];
        }
        
    }

 

面试题:合并2个有序数组(leetcode88)

标签:nbsp   div   pre   空间   col   有序   style   leetcode   span   

原文地址:https://www.cnblogs.com/starcrm/p/10695347.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!