天天看点

Leetcode 306. 累加数 (迭代方法,每次选取字符串两个数,不断向后跌带,直到迭代到最后)

Leetcode 306. 累加数 (迭代方法,每次选取字符串两个数,不断向后跌带,直到迭代到最后)
class Solution:
    def isAdditiveNumber(self, num: str) -> bool:
        n = len(num)
        for i, j in itertools.combinations(range(1, n), 2):
            # 第一个数是从0到i-1, 第二个数是i到j-1
            a, b = num[:i], num[i:j]
            if a != str(int(a)) or b != str(int(b)):
                continue
            while j < n:
                c = str(int(a) + int(b))
                if not num.startswith(c, j): break
                j += len(c)
                a, b = b, c
            if j == n: return True
        return False