LeetCode刷题日记2022-1-10/306. 累加数-循环判断

题目描述

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

一个有效的 累加序列 必须 至少 包含 3 个数。除了最开始的两个数以外，字符串中的其他数都等于它之前两个数相加的和。

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

说明：累加序列里的数 不会 以 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）组成

题解思路

当我们确定了第一个数字 以及第二个数字之后 就可以判断所给定的字符串是否是一个累加序列

因此我们遍历第一个数字 和 第二个数字的所有组合即可

题解代码

class Solution:
	def isAdditiveNumber(self, num: str) -> bool:
        def check(i,j):
            a=int(num[:i+1])
            b=int(num[i+1:j+1])
            #如果存在前置为0则不合法
            if (i>0 and num[0]=='0') or (j>i+1 and num[i+1]=='0'):
                return False
            #进入循环判断序列是否合法
            while True:
                if j==n-1:
                    return True
                c=a+b
                c_str=str(c)
                k=j+len(c_str)
                if k>n:
                    return False
                if c!=int(num[j+1:k+1]):
                    return False
                else:
                    i,j=j,k
                    a,b=b,c
            return True
		
		n=len(num)
		for i in range(n-2):
			for j in range(i+1,n-1):
				if check(i,j):
					return True
		return False

其中i,j,k分别为第1个数字，第二个数字 第三个数字的结尾