LeetCode842. 将数组拆分成斐波那契序列（回溯+剪枝）

1、题目描述

https://leetcode-cn.com/problems/split-array-into-fibonacci-sequence/

给定一个数字字符串 S，比如 S = "123456579"，我们可以将它分成斐波那契式的序列 [123, 456, 579]。

另外，请注意，将字符串拆分成小块时，每个块的数字一定不要以零开头，除非这个块是数字 0 本身。

返回从 S 拆分出来的任意一组斐波那契式的序列块，如果不能拆分则返回 []。

输入："123456579"
输出：[123,456,579]

输入: "11235813"
输出: [1,1,2,3,5,8,13]

输入: "112358130"
输出: []
解释: 这项任务无法完成。

输入："0123"
输出：[]
解释：每个块的数字不能以零开头，因此 "01"，"2"，"3" 不是有效答案。

输入: "1101111"
输出: [110, 1, 111]
解释: 输出 [11,0,11,11] 也同样被接受。

1. 1 <= S.length <= 200
2. 字符串 S 中只含有数字。

2、代码详解

回溯法+剪枝

https://leetcode-cn.com/problems/split-array-into-fibonacci-sequence/solution/javahui-su-suan-fa-tu-wen-xiang-jie-ji-b-vg5z/

from typing import List


class Solution:
    def splitIntoFibonacci(self, S: str) -> List[int]:

        def backtrack(s, res, index):
            # 边界条件判断，如果截取完了，并且res长度大于等于3，表示找到了一个组合
            if len(res) >= 3 and index == len(s):
                return True

            for i in range(index, len(s)):
                # 两位以上的数字不能以0开头
                # 如果只是0 就不用跳出
                # 将字符串拆分成小块时，每个块的数字一定不要以零开头，除非这个块是数字 0 本身
                if s[index] == '0' and i > index:
                    break
                # 截取字符串转化为数字
                num = subDight(s, index, i+1)
                # 如果截取的数字大于int的最大值，则终止截取
                if num > 2 ** 31 - 1:
                    break
                size = len(res)
                # 如果截取的数字大于res中前两个数字的和，
                # 说明这次截取的太大，直接终止，因为后面越截取越大
                if size >= 2 and num > (res[size-1] + res[size-2]):
                    break
                if size <= 1 or num == res[size-1] + res[size-2]:
                    res.append(num)  # 把数字num添加到集合res中
                    # 如果找到了就直接返回
                    if backtrack(s, res, i+1):
                        return True
                    # 如果没找到，就会走回溯这一步，然后把上一步添加到集合res中的数字给移除掉
                    res.pop(-1)
            return False

        # 相当于截取字符串S中的子串然后转换为十进制数字
        def subDight(dight, start, end):
            num = 0
            for i in range(start, end):
                num = num*10 + int(dight[i])
            return num


        ans = list()
        backtrack(S, ans, 0)
        return ans

S = "11235813"
s = Solution()
print(s.splitIntoFibonacci(S))  # [1, 1, 2, 3, 5, 8, 13]