Re-implement merge sort function. (#888)

codes/go/chapter_sorting/merge_sort.go

@@ -5,35 +5,37 @@
 package chapter_sorting
 
 /* 合并左子数组和右子数组 */
-// 左子数组区间 [left, mid]
-// 右子数组区间 [mid + 1, right]
 func merge(nums []int, left, mid, right int) {
-	// 初始化辅助数组 借助 copy 模块
+	// 左子数组区间 [left, mid], 右子数组区间 [mid+1, right]
+	// 创建一个临时数组 tmp ，用于存放合并后的结果
 	tmp := make([]int, right-left+1)
-	for i := left; i <= right; i++ {
-		tmp[i-left] = nums[i]
-	}
-	// 左子数组的起始索引和结束索引
-	leftStart, leftEnd := left-left, mid-left
-	// 右子数组的起始索引和结束索引
-	rightStart, rightEnd := mid+1-left, right-left
-	// i, j 分别指向左子数组、右子数组的首元素
-	i, j := leftStart, rightStart
-	// 通过覆盖原数组 nums 来合并左子数组和右子数组
-	for k := left; k <= right; k++ {
-		// 若“左子数组已全部合并完”，则选取右子数组元素，并且 j++
-		if i > leftEnd {
-			nums[k] = tmp[j]
-			j++
-			// 否则，若“右子数组已全部合并完”或“左子数组元素 <= 右子数组元素”，则选取左子数组元素，并且 i++
-		} else if j > rightEnd || tmp[i] <= tmp[j] {
-			nums[k] = tmp[i]
+	// 初始化左子数组和右子数组的起始索引
+	i, j, k := left, mid+1, 0
+	// 当左右子数组都还有元素时，比较并将较小的元素复制到临时数组中
+	for i <= mid && j <= right {
+		if nums[i] <= nums[j] {
+			tmp[k] = nums[i]
 			i++
-			// 否则，若“左右子数组都未全部合并完”且“左子数组元素 > 右子数组元素”，则选取右子数组元素，并且 j++
 		} else {
-			nums[k] = tmp[j]
+			tmp[k] = nums[j]
 			j++
 		}
+		k++
+	}
+	// 将左子数组和右子数组的剩余元素复制到临时数组中
+	for i <= mid {
+		tmp[k] = nums[i]
+		i++
+		k++
+	}
+	for j <= right {
+		tmp[k] = nums[j]
+		j++
+		k++
+	}
+	// 将临时数组 tmp 中的元素复制回原数组 nums 的对应区间
+	for k := 0; k < len(tmp); k++ {
+		nums[left+k] = tmp[k]
 	}
 }