合併兩個有序陣列的golang實現
阿新 • • 發佈:2018-12-07
給定兩個有序整數陣列 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) }