合並兩個有序數組的golang實現


給定兩個有序整數數組 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)
}

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM