这个题还真挺简单的,值得记录的是这个题要求的数的范围在int内,但python的int函数其实是可以超过int范围的,所以咱们需要判断一下在(1<<31)-1范围内,另外,可以利用这一点,只搜索长度10以内的子串,是一个小的剪枝技巧。
第二个点也是一个剪枝技巧,由于咱们需要先找到两个数,如果第一个数开头是0,那就不用搜索第二个数了。
代码如下:
class Solution:
def splitIntoFibonacci(self, S: str) -> List[int]:
mx = int(len(S)/3)+1
int_max = (1<<31)-1
# 给定前两个数,遍历并判断
def find(l, i, S):
while i < len(S):
ret = str(int(l[-1]) + int(l[-2]))
if i+len(ret)-1 < len(S) and S[i:i+len(ret)] == ret and int(ret) <= int_max:
l.append(ret)
i += len(ret)
else: return
return l
# 穷尽搜索可能的前两个数
for i in range(1, min(10, mx+1)):
first = S[:i]
if first[0] == '0' and len(first)>1:
continue
for j in range(i+1, min(i+10, i+mx+1)):
second = S[i:j]
if second[0] == '0' and len(second)>1:
continue
ret = find([first, second], j, S)
if ret: return ret
return []