【LeetCode】842. Split Array into Fibonacci Sequence 解题报告（Python & C++）

原创

负雪明烛 2021-07-14 10:55:12 ©著作权

©著作权归作者所有：来自51CTO博客作者负雪明烛的原创作品，请联系作者获取转载授权，否则将追究法律责任

作者：负雪明烛
id： fuxuemingzhu
个人博客： http://fuxuemingzhu.cn/

题目描述

Given a string S of digits, such as S = "123456579", we can split it into a Fibonacci-like sequence [123, 456, 579].

Formally, a Fibonacci-like sequence is a list F of non-negative integers such that:

0 <= F[i] <= 2^31 - 1, (that is, each integer fits a 32-bit signed integer type);
F.length >= 3;
and F[i] + F[i+1] = F[i+2] for all 0 <= i < F.length - 2.

Also, note that when splitting the string into pieces, each piece must not have extra leading zeroes, except if the piece is the number 0 itself.

Return any Fibonacci-like sequence split from S, or return [] if it cannot be done.

Example 1:

Input: "123456579"
Output: [123,456,579]

Example 2:

Input: "11235813"
Output: [1,1,2,3,5,8,13]

Example 3:

Input: "112358130"
Output: []
Explanation: The task is impossible.

Example 4:

Input: "0123"
Output: []
Explanation: Leading zeroes are not allowed, so "01", "2", "3" is not valid.

Example 5:

Input: "1101111"
Output: [110, 1, 111]
Explanation: The output [11, 0, 11, 11] would also be accepted.

Note:

1 <= S.length <= 200
S contains only digits.

题目大意

给出了一个有0-9数字组成的纯数字字符串。判断能否组成所谓的费布拉奇数列。注意这个题注重点在不管你几位数字去划分，只要满足后面的数字等于前两个的和即可。最终要返回的是任何一个组合即可。

解题方法

按照Tag说就是快啊，这个题和306. Additive Number一个一模一样啊，306题是要返回True和False，这个是要求返回具体的一个例子。

因为只要判断能否构成即可，所以不需要res数组保存结果。回溯法仍然是对剩余的数字进行切片，看该部分切片能否满足条件。剪枝的方法是判断数组是否长度超过3，如果超过那么判断是否满足费布拉奇数列的规则。不超过3或者已经满足的条件下继续进行回溯切片。最后当所有的字符串被切片完毕，要判断下数组长度是否大于等于3，这是题目要求。

因为题目要求返回任意一个就好了，因此，只要找到一个满足条件的，那么就返回True，再结束循环就好了。所以整个题都是在306的基础上做出来的。

第一遍提交的时候出了个错，第一遍竟然没看出来：

输入："539834657215398346785398346991079669377161950407626991734534318677529701785098211336528511"
输出：[539834657,21,539834678,539834699,1079669377,1619504076,2699173453,4318677529,7017850982,11336528511]

仔细一想，是最后的数字超过了2**31，python不会报错。。如果是c++或者java应该还是挺容易看出来的。

代码如下：

class Solution(object):
    def splitIntoFibonacci(self, S):
        """
        :type S: str
        :rtype: List[int]
        """
        res = []
        self.dfs(S, [], res)
        return res
        
    def dfs(self, num_str, path, res):
        if len(path) >= 3 and  path[-1] != path[-2] + path[-3]:
            return False
        if not num_str and len(path) >= 3:
            res.extend(path)
            return True
        for i in range(len(num_str)):
            curr = num_str[:i+1]
            if (curr[0] == '0' and len(curr) != 1) or int(curr) >= 2**31:
                continue
            if self.dfs(num_str[i+1:], path + [int(curr)], res):
                return True
        return False

二刷使用C++代码如下：

class Solution {
public:
    vector<int> splitIntoFibonacci(string S) {
        vector<int> path;
        helper(S, path, 0);
        return path;
    }
    // [start, S.size())
    bool helper(string& num, vector<int>& path, int start) {
        if (start >= num.size() && path.size() >= 3)
            return true;
        for (int i = 1; start + i <= num.size(); i++) {
            if (num[start] == '0' && i > 1) break;
            long long subll = stoll(num.substr(start, i));
            if (subll > INT_MAX) 
                return false;
            if (path.size() >= 2 && subll > path[path.size() - 1] + path[path.size() - 2])
                return false;
            if (path.size() <= 1 || subll == path[path.size() - 1] + path[path.size() - 2]) {
                path.push_back((int)subll);
                if (helper(num, path, start + i)) {
                    return true;
                }
                path.pop_back();
            }
        }
        return false;
    }
};