swift 算法 简单17.合并两个有序数组

知识共享许可协议 Creative Commons

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

解法:

   func merge(_ nums1: inout [Int], _ m: Int, _ nums2: [Int], _ n: Int) {
        var m = m,n = n
        var pos = m + n - 1; //两个数组的下标长度
        while (n > 0) {//遍历第二个数组
            if (m > 0 && nums1[m - 1] > nums2[n - 1]) { //数组1中的最后一个数大于数组2中的最后一个元素
                //数组1中的最后一个数 移到数组1中的最后一个位置
                nums1[pos] = nums1[m-1];
                pos -= 1
                m -= 1
            }else {
                //数组2中的最后一个数 移到数组1中的最后一个位置
                nums1[pos] = nums2[n-1];
                pos -= 1
                n -= 1
            }
        }

    }

猜你喜欢

转载自blog.csdn.net/huanglinxiao/article/details/91946479