add insertion 插入排序

Author: colynn

insertion_sort.go

@@ -0,0 +1,19 @@
+package algorithm
+
+// InsertionSort ..
+// 算法步骤：
+// 将第一待排序序列第一个元素看做一个有序序列，把第二个元素到最后一个元素当成是未排序序列。
+// 从头到尾依次扫描未排序序列，将扫描到的每个元素插入有序序列的适当位置。
+//（如果待插入的元素与有序序列中的某个元素相等，则将待插入元素插入到相等元素的后面。）
+func InsertionSort(arr []int) []int {
+	for i := range arr {
+		preIndex := i - 1
+		current := arr[i]
+		for preIndex >= 0 && arr[preIndex] > current {
+			arr[preIndex+1] = arr[preIndex]
+			preIndex--
+		}
+		arr[preIndex+1] = current
+	}
+	return arr
+}

insertion_sort_test.go

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