LeetCode 306. Additive Number

LeetCode 306. Additive Number

Description

Additive number is a string whose digits can form additive sequence.

A valid additive sequence should contain at least three numbers. Except for the first two numbers, each subsequent number in the sequence must be the sum of the preceding two.

Given a string containing only digits '0'-'9', write a function to determine if it's an additive number.

Note: Numbers in the additive sequence cannot have leading zeros, so sequence 1, 2, 03 or 1, 02, 3 is invalid.

Example 1:

Input: "112358"
Output: true 
Explanation: The digits can form an additive sequence: 1, 1, 2, 3, 5, 8. 
             1 + 1 = 2, 1 + 2 = 3, 2 + 3 = 5, 3 + 5 = 8

Example 2:

Input: "199100199"
Output: true 
Explanation: The additive sequence is: 1, 99, 100, 199. 
             1 + 99 = 100, 99 + 100 = 199

描述

累加数是一个字符串，组成它的数字能够造成累加序列。

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

给定一个只包含数字 '0'-'9' 的字符串，编写一个算法来判断给定输入是不是累加数。

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

思路

• 这道题可使用深度优先搜索进行回溯.
• 咱们使用深度优先搜索，找到全部可能拆分给定字符串的方式，而后咱们判断当前的拆分方式是否能构成累加数，若是能够，咱们用res记录为True，不然为False.
• __valid函数用于判断当前的组合方式是否能构成累加数，注意：累加数至少须要三个.

# -*- coding: utf-8 -*-
# @Author:             何睿
# @Create Date:        2019-02-11 20:50:12
# @Last Modified by:   何睿
# @Last Modified time: 2019-02-11 21:25:41


class Solution:
    def isAdditiveNumber(self, num: 'str') -> 'bool':
        # 根据题意，累计加数至少有三个
        if len(num) < 3: return False
        self.res = False
        # 深度优先搜索，遍历全部可能的解
        self.__dfs(0, num, [])
        return self.res

    def __dfs(self, start, num, coms):
        # 递归结束条件，当num中没有数字时，检查当前组合是否知足条件
        if start == len(num):
            # 若是当前组合合法，咱们将self.res置为True
            if self.__valid(coms): self.res = True
            return
        # 记录起始位置
        index = start
        while index < len(num):
            # 若是当前数字的起始数字是"0'退出循环（注意单独一个'0'自己是合法的）
            if num[start] == "0" and index != start: break
            # 若是当前的组合已经有了至少3个数，咱们检查前面的全部数是不是累加数
            # 若是不是咱们退出循环，表示当前的分支不用再查找，减小时间
            if len(coms) > 2 and not self.__valid(coms): break
            # 递归遍历分支
            self.__dfs(index + 1, num, coms + [num[start:index + 1]])
            index += 1

    def __valid(self, coms):
        # 若是一共都没有三个数，返回False
        if len(coms) < 3: return False
        for i in range(len(coms) - 2):
            # 只要有一个不知足累加数的条件，返回False
            if int(coms[i]) + int(coms[i + 1]) != int(coms[i + 2]):
                return False
        return True

源代码文件在这里.
©本文首发于何睿的博客，欢迎转载，转载需保留文章来源，做者信息和本声明.