finish Problem 38

Author: aQua

Algorithms/0038.count-and-say/README.md

@@ -31,8 +31,9 @@ Output: "1211"
 ```
 
 ## 解题思路
+利用[]byte记录上一次出现的次数和元素。
 
+详见注释
 
 ## 总结
-
-
+使用辅助结果，比如切片或者映射记录信息，可以极大地加快程序

Algorithms/0038.count-and-say/count-and-say.go

@@ -1,40 +1,33 @@
 package Problem0038
 
-import (
-	"strconv"
-)
-
 func countAndSay(n int) string {
-	if n == 1 {
-		return "1"
-	}
-	return say(countAndSay(n - 1))
-}
+	buf := []byte{'1'}
 
-func say(s string) string {
-	res := ""
-
-	words := split(s)
-	for _, w := range words {
-		res += strconv.Itoa(len(w)) + w[0:1]
+	for n > 1 {
+		buf = say(buf)
+		n--
 	}
 
-	return res
+	return string(buf)
 }
 
-func split(s string) []string {
-	res := []string{}
-	i, j := 0, 1
+func say(buf []byte) []byte {
+	// res 长度不会超过 buf 的两倍，所以，可以事先指定容量，加快append的速度
+	res := make([]byte, 0, len(buf)*2)
 
-	for j < len(s) {
-		if s[i] != s[j] {
-			res = append(res, s[i:j])
-			i = j
+	i, j := 0, 1
+	for i < len(buf) {
+		// 利用 j ，找到下一个不同的元素
+		for j < len(buf) && buf[j] == buf[i] {
+			j++
 		}
-		j++
-	}
 
-	res = append(res, s[i:])
+		// res 中 res[i] 表示 res[i+1] 的个数，i 为0,2,4,6,...
+		res = append(res, byte(j-i+'0'), buf[i])
+
+		// 移动 i 到 j
+		i = j
+	}
 
 	return res
 }

Algorithms/0038.count-and-say/count-and-say_test.go

@@ -79,21 +79,3 @@ func Test_Problem0038(t *testing.T) {
 		ast.Equal(a.one, countAndSay(p.one), "输入:%v", p)
 	}
 }
-
-func Test_say(t *testing.T) {
-	ast := assert.New(t)
-
-	ast.Equal("1211", say("21"), "没有把 21 说成 1211")
-}
-
-func Test_split(t *testing.T) {
-	ast := assert.New(t)
-
-	actual := split("333221")
-	expected := []string{"333", "22", "1"}
-	ast.Equal(expected, actual, "没能分隔字符串")
-
-	actual = split("11")
-	expected = []string{"11"}
-	ast.Equal(expected, actual, "没能分隔字符串")
-}