842 finish

Author: aQua

.vscode/bookmarks.json

@@ -8,7 +8,7 @@
 		{
 			"fsPath": "$ROOTPATH$/Algorithms/0842.split-array-into-fibonacci-sequence/split-array-into-fibonacci-sequence_test.go",
 			"bookmarks": [
-				38
+				43
 			]
 		}
 	]

Algorithms/0842.split-array-into-fibonacci-sequence/split-array-into-fibonacci-sequence.go

@@ -15,9 +15,8 @@ func splitIntoFibonacci(s string) []int {
 			}
 			a, _ := strconv.Atoi(s[:i])
 			b, _ := strconv.Atoi(s[i:j])
-			res := make([]int, 2, len(s))
-			res[0] = a
-			res[1] = b
+			res := make([]int, 0, len(s))
+			res = append(res, a, b)
 			if find(a, b, s[j:], &res) {
 				return res
 			}
@@ -27,23 +26,24 @@ func splitIntoFibonacci(s string) []int {
 }
 
 func find(a, b int, s string, res *[]int) bool {
-	if s == "" {
-		return true
-	}
+	for len(s) > 0 {
+		c := a + b
+		if c > 1<<31-1 {
+			return false
+		}
+		cs := strconv.Itoa(c)
+		csl := len(cs)
 
-	c := a + b
-	if c > 1<<31-1 {
-		return false
-	}
-	cs := strconv.Itoa(c)
-	csl := len(cs)
+		if len(s) < csl || s[:csl] != cs {
+			return false
+		}
 
-	if len(s) >= csl && s[:csl] == cs {
 		*res = append(*res, c)
-		return find(b, c, s[csl:], res)
+		a, b = b, c
+		s = s[csl:]
 	}
 
-	return false
+	return true
 }
 
 func isLeadingZero(s string) bool {

Algorithms/0842.split-array-into-fibonacci-sequence/split-array-into-fibonacci-sequence_test.go

@@ -14,15 +14,20 @@ var tcs = []struct {
 }{
 
 	{
-		"539834657215398346785398346991079669377161950407626991734534318677529701785098211336528511",
-		nil,
+		"1320581321313221264343965566089105744171833277577",
+		[]int{13205, 8, 13213, 13221, 26434, 39655, 66089, 105744, 171833, 277577},
 	},
 
 	{
 		"123456579",
 		[]int{123, 456, 579},
 	},
 
+	{
+		"539834657215398346785398346991079669377161950407626991734534318677529701785098211336528511",
+		nil,
+	},
+
 	{
 		"11235813",
 		[]int{1, 1, 2, 3, 5, 8, 13},