842 accepted

Author: aQua

.vscode/bookmarks.json

@@ -0,0 +1,15 @@
+{
+	"onDidClearBookmarkEmitter": {},
+	"onDidClearAllBookmarksEmitter": {},
+	"onDidAddBookmarkEmitter": {},
+	"onDidRemoveBookmarkEmitter": {},
+	"onDidUpdateBookmarkEmitter": {},
+	"bookmarks": [
+		{
+			"fsPath": "$ROOTPATH$/Algorithms/0842.split-array-into-fibonacci-sequence/split-array-into-fibonacci-sequence_test.go",
+			"bookmarks": [
+				38
+			]
+		}
+	]
+}

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

@@ -1,6 +1,58 @@
 package problem0842
 
-func splitIntoFibonacci(s string) []int {
+import (
+	"strconv"
+)
 
+func splitIntoFibonacci(s string) []int {
+	for i := 1; i <= len(s)/2; i++ {
+		if isLeadingZero(s[:i]) {
+			break
+		}
+		for j := i + 1; max(i, j-i) <= len(s)-j; j++ {
+			if isLeadingZero(s[i:j]) {
+				break
+			}
+			a, _ := strconv.Atoi(s[:i])
+			b, _ := strconv.Atoi(s[i:j])
+			res := make([]int, 2, len(s))
+			res[0] = a
+			res[1] = b
+			if find(a, b, s[j:], &res) {
+				return res
+			}
+		}
+	}
 	return nil
 }
+
+func find(a, b int, s string, res *[]int) bool {
+	if s == "" {
+		return true
+	}
+
+	c := a + b
+	if c > 1<<31-1 {
+		return false
+	}
+	cs := strconv.Itoa(c)
+	csl := len(cs)
+
+	if len(s) >= csl && s[:csl] == cs {
+		*res = append(*res, c)
+		return find(b, c, s[csl:], res)
+	}
+
+	return false
+}
+
+func isLeadingZero(s string) bool {
+	return len(s) > 1 && s[0] == '0'
+}
+
+func max(a, b int) int {
+	if a > b {
+		return a
+	}
+	return b
+}

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

@@ -13,6 +13,11 @@ var tcs = []struct {
 	ans []int
 }{
 
+	{
+		"539834657215398346785398346991079669377161950407626991734534318677529701785098211336528511",
+		nil,
+	},
+
 	{
 		"123456579",
 		[]int{123, 456, 579},
@@ -25,17 +30,17 @@ var tcs = []struct {
 
 	{
 		"112358130",
-		[]int{},
+		nil,
 	},
 
 	{
 		"0123",
-		[]int{},
+		nil,
 	},
 
 	{
 		"1101111",
-		[]int{110, 1, 111},
+		[]int{11, 0, 11, 11},
 	},
 
 	// 可以有多个 testcase