Leetcode 842. 将数组拆分成斐波那契序列

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

形式上,斐波那契式序列是一个非负整数列表 F,且满足:

  • 0 <= F[i] <= 2^31 - 1,(也就是说,每个整数都符合 32 位有符号整数类型);
  • F.length >= 3
  • 对于所有的0 <= i < F.length - 2,都有 F[i] + F[i+1] = F[i+2] 成立。

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

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

 

示例 1:

输入:"123456579"

输出:[123,456,579]

示例 2:

输入: "11235813"

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

示例 3:

输入: "112358130"

输出: []

解释: 这项任务无法完成。

示例 4:

输入:"0123"

输出:[]

解释:每个块的数字不能以零开头,因此 "01","2","3" 不是有效答案。

示例 5:

输入: "1101111"

输出: [110, 1, 111]

解释: 输出 [11,0,11,11] 也同样被接受。

 

提示:

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

**难度**: Medium

**标签**: 贪心算法、 字符串、 回溯算法、


# -*- coding: utf-8 -*-
# @Author  : LG

"""
执行用时:148 ms, 在所有 Python3 提交中击败了35.57% 的用户
内存消耗:13.7 MB, 在所有 Python3 提交中击败了36.81% 的用户

解题思路:
    回溯
    具体实现见代码注释
"""
class Solution:
    def splitIntoFibonacci(self, S: str) -> List[int]:

        def backtrack(S, current):
            if S == '' and len(current) > 2:    # 字符串均处理完,且当前序列长度大于2, 返回最终结果
                return True

            if S == '': # 当字符串处理完时,跳出
                return

            for i in range(1, len(S)+1): # 遍历当前字符串
                if (S[0] == '0' and i == 1) or (S[0] != '0'):   # 排除以0 开头的非0数, 如 01 02 等
                    if int(S[:i]) < (2**31-1) and (len(current) < 2 or int(S[:i]) == int(current[-1]) + int(current[-2])):  # 数字限制;长度判断,如长度小于2,直接添加,如长度大于2,需判断和
                        current.append(S[:i])
                        if backtrack(S[i:], current):
                            return current
                        current.pop()

        result = backtrack(S, [])
        if result:
            return result
        else:
            return []