add merge sort

Author: colynn

docs/5.merge_sort.md

@@ -5,4 +5,11 @@
 
 作为一种典型的分而治之思想的算法应用，归并排序的实现由两种方法：
 * 自上而下的递归（所有递归的方法都可以用迭代重写，所以就有了第2种方法）；
-* 自下而上的迭代； 
+* 自下而上的迭代； 
+
+## 算法步骤
+1. 申请空间，使其大小为两个已经排序序列之和，该空间用来存放合并后的序列；
+2. 设定两个指针，最初位置分别为两个已经排序序列的起始位置；
+3. 比较两个指针所指向的元素，选择相对小的元素放入到合并空间，并移动指针到下一位置；
+4. 重复步骤3直到某一指针达到序列尾；
+5. 将另一序列剩下的所有元素直接复制到合并序列列尾。

insertion_sort.go

@@ -1,7 +1,5 @@
 package algorithm
 
-import "fmt"
-
 // InsertionSort ..
 // No.3
 // 插入排序（稳定）算法步骤：
@@ -13,12 +11,12 @@ func InsertionSort(arr []int) []int {
 	for i := range arr {
 		preIndex := i - 1
 		current := arr[i]
-		fmt.Printf("current: %v\n", current)
+		// fmt.Printf("current: %v\n", current)
 		for preIndex >= 0 && arr[preIndex] > current {
 			arr[preIndex+1] = arr[preIndex]
 			preIndex--
 		}
-		fmt.Printf("finally i: %v, current: %v, preIndex: %v\n", i, current, preIndex)
+		// fmt.Printf("finally i: %v, current: %v, preIndex: %v\n", i, current, preIndex)
 		arr[preIndex+1] = current
 	}
 	return arr

merge_sort.go

@@ -0,0 +1,36 @@
+package algorithm
+
+// MergeSort ..
+// No.5
+func mergeSort(arr []int) []int {
+	length := len(arr)
+	if length < 2 {
+		return arr
+	}
+	middle := length / 2
+	left := arr[0:middle]
+	right := arr[middle:]
+	return merge(mergeSort(left), mergeSort(right))
+}
+
+func merge(left []int, right []int) []int {
+	var result []int
+	for len(left) != 0 && len(right) != 0 {
+		if left[0] < right[0] {
+			result = append(result, left[0])
+			left = left[1:]
+		} else {
+			result = append(result, right[0])
+			right = right[1:]
+		}
+	}
+	for len(left) != 0 {
+		result = append(result, left[0])
+		left = left[1:]
+	}
+	for len(right) != 0 {
+		result = append(result, right[0])
+		right = right[1:]
+	}
+	return result
+}

merge_sort_test.go

@@ -0,0 +1,16 @@
+package algorithm
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestMergeSort(t *testing.T) {
+	arr := []int{3, 2, 0, 5, 10, 1}
+	arrSort := mergeSort(arr)
+	fmt.Printf("arr after sored: %v\n", arrSort)
+	assert.Equal(t, 10, arrSort[5])
+	assert.Equal(t, 5, arrSort[4])
+}