累加数

183 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 16 天,点击查看活动详情

累加数 是一个字符串,组成它的数字可以形成累加序列。

一个有效的 累加序列 必须 至少 包含 3 个数。除了最开始的两个数以外,序列中的每个后续数字必须是它之前两个数字之和。

给你一个只包含数字 '0'-'9' 的字符串,编写一个算法来判断给定输入是否是 累加数 。如果是,返回 true ;否则,返回 false 。

说明:累加序列里的数,除数字 0 之外,不会 以 0 开头,所以不会出现 1, 2, 03 或者 1, 02, 3 的情况。

示例 1:

输入:"112358" 输出:true 解释:累加序列为: 1, 1, 2, 3, 5, 8 。1 + 1 = 2, 1 + 2 = 3, 2 + 3 = 5, 3 + 5 = 8

示例 2:

输入:"199100199" 输出:true 解释:累加序列为: 1, 99, 100, 199。1 + 99 = 100, 99 + 100 = 199

提示:

1 <= num.length <= 35
num 仅由数字(0 - 9)组成

进阶: 你计划如何处理由过大的整数输入导致的溢出?

解题思路

使用递归每次将串化成三份 比如"112358"可以分成’1’,‘1’,‘2358’;尝试前两个数的和能否在后面的’2358’中找到

因为’2358’中的‘2’=1+1;找到后用1+2=3判断在’‘358’'中是否包含3。依次类推

但对于 “199100199”,第一次分是’1’,‘9’,‘9100199’,明显1+9在’9100199’中找不到那么就得重新将其分成三份 1, 99, 100199。 这是就可以递推了

代码

    def isAdditiveNumber(self, num: str) -> bool:
        def helper(a, b):
            i, j, res, one = len(a) - 1, len(b) - 1, [], 0
            while i >= 0 or j >= 0:
                curA = curB = 0
                if i >= 0:
                    curA = int(a[i])
                    i -= 1
                if j >= 0:
                    curB = int(b[j])
                    j -= 1
                cur = curA + curB + one
                if cur >= 10:
                    cur %= 10
                    one = 1
                else:
                    one = 0
                res.append(str(cur))
            if one:
                res.append("1")
            return "".join(res[::-1])
        
        def dfs(p, q):
            last, first, second = 0, p, q
            while second < len(num):
                if (num[last] == '0' and first > last + 1) or (num[first] == '0' and second > first + 1):
                    return False
                s = helper(num[last:first], num[first:second])
                if num[second:second+len(s)] != s:
                    return False
                last, first, second = first, second, second + len(s)
            return True
        
        for i in range(1, len(num) - 1):
            for j in range(i + 1, len(num)):
                if dfs(i, j):
                    return True
        return False