合并两个有序数组的golang实现

Posted 2021-01-25 timliudream

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

由题目中我们可以知道其实就是把nums1的前m项和nums2的前n项合并，并放进nums1中

因为nums1要变动，所以我们拷贝nums1到一个临时变量中

    //把nums1复制到temp中
    temp := make([]int, m)
    copy(temp, nums1)

然后就是对比temp与nums2对应的值，比较哪个比较小，小的那个就插入nums1中

    t, j := 0, 0 //t为temp的索引，j为nums2的索引
    for i := 0; i < len(nums1); i++ {
        //比较nums2与temp对应值的大小，小的那个就放进nums1中
        if nums2[j] <= temp[t] {
            nums1[i] = nums2[j]
            j++
        } else {
            nums1[i] = temp[t]
            t++
        }
    }

但是这样写，总有一个会先用完，当用完的时候，索引++会导致

index out of range

所以要改进一下代码，整体代码：

// 本题的要求是，把nums1的前m项和nums2的前n项合并，放入nums1中。
func merge(nums1 []int, m int, nums2 []int, n int) {
    //把nums1复制到temp中
    temp := make([]int, m)
    copy(temp, nums1)

    t, j := 0, 0 //t为temp的索引，j为nums2的索引
    for i := 0; i < len(nums1); i++ {
        //当t大于temp的长度，那就是说temp全部放进去了nums1中，那剩下的就是放nums2剩余的值了
        if t >= len(temp) {
            nums1[i] = nums2[j]
            j++
            continue
        }
        //当j大于nums2的长度的时候，那就是说明nums2全部都放进去了nums1中，那剩下的就是放temp剩余的值了
        if j >= n {
            nums1[i] = temp[t]
            t++
            continue
        }
        //比较nums2与temp对应值的大小，小的那个就放进nums1中
        if nums2[j] <= temp[t] {
            nums1[i] = nums2[j]
            j++
        } else {
            nums1[i] = temp[t]
            t++
        }
    }
    fmt.Println(nums1)
}