swift算法：合并两个有序数组

1、描述

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

说明：

 1）初始化nums1 和 nums2的元素数量分别为 m 和 n

2）你可以假设 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]

例2：

        输入：nums1 = [-1, 0, 0, 3, 3, 3, 0, 0, 0],  m = 6

                    nums2 = [1, 2, 2],    n = 3

        输出：[-1, 0, 0, 1, 2, 2, 3, 3, 3]

 

 

2、算法

解法一：合并后排序

思想：将nums2的n个数合并到nums1的m～n-1下标位置，再对nums1排序

时间复杂度：O(n)

func merge(_ nums1: inout [Int], _ m: Int, _ nums2: [Int], _ n: Int) {
        /*
         合并后排序
         */
        nums1 = [Int](nums1[0..<m])
        for num in nums2 {
            nums1.append(num)
        }
        nums1 = nums1.sorted()
        print(nums1)
    }

解法二：从前往后排序

思想：比较两个数组的第一个元素，然后加入新的数组，移除旧数组中的，最后判断是否有剩下的，若有全部加入新数组，最后将新数组赋值给nums1

时间复杂度：O(n)

func merge(_ nums1: inout [Int], _ m: Int, _ nums2: [Int], _ n: Int) {
        /*
         从前往后排序
         */
        if m==0 {
            nums1 = nums2
        }
        var num = [Int]()
        var newNum1 = [Int](nums1[0..<m])
        var nums2 : [Int] = nums2
        while !newNum1.isEmpty && !nums2.isEmpty {
            if newNum1.first! <= nums2.first!{
                num.append(newNum1.first!)
                newNum1.removeFirst()
            }else{
                num.append(nums2.first!)
                nums2.removeFirst()
            }
            if newNum1.isEmpty {
                num.append(contentsOf: nums2)
                nums1 = num
            }
            if nums2.isEmpty{
                num.append(contentsOf: newNum1)
                nums1 = num
            }
        }
        print(nums1)
        
    }
   

解法三：从后往前排序

思想：利用双指针，将最大的数加入nums1，依次循环

时间复杂度

func merge(_ nums1: inout [Int], _ m: Int, _ nums2: [Int], _ n: Int) {
        //双指针 从后往前
        var p1 = m-1
        var p2 = n-1
        var p = m+n-1
        while p1>=0 && p2>=0 {
            if  nums1[p1] <= nums2[p2]{
                nums1[p] = nums2[p2]
                p2 -= 1
                p -= 1
            }else{
                nums1[p] = nums1[p1]
                p1 -= 1
                p -= 1
                
            }
        }
        while p2 >= 0 {
            nums1[p] = nums2[p2]
            p -= 1
            p2 -= 1
        }
    }