LeetCode 842. 将数组拆分成斐波那契序列

题目描述

给定一个数字字符串 S，比如 S = "123456579"，我们可以将它分成斐波那契式的序列 [123, 456, 579]。

形式上，斐波那契式序列是一个非负整数列表 F，且满足：

• 0 <= F[i] <= 2^31 - 1，（也就是说，每个整数都符合 32 位有符号整数类型）；
• F.length >= 3；
• 对于所有的0 <= i < F.length - 2，都有 F[i] + F[i+1] = F[i+2] 成立。

另外，请注意，将字符串拆分成小块时，每个块的数字一定不要以零开头，除非这个块是数字 0 本身。

返回从 S 拆分出来的所有斐波那契式的序列块，如果不能拆分则返回 []。

示例 1：

输入："123456579"
输出：[123,456,579]

示例 2：

输入: "11235813"
输出: [1,1,2,3,5,8,13]

示例 3：

输入: "112358130"
输出: []
解释: 这项任务无法完成。

示例 4：

输入："0123"
输出：[]
解释：每个块的数字不能以零开头，因此 "01"，"2"，"3" 不是有效答案。

示例 5：

输入: "1101111"
输出: [110, 1, 111]
解释: 输出 [11,0,11,11] 也同样被接受。

提示：

1. 1 <= S.length <= 200
2. 字符串 S 中只含有数字。

问题分析

此题和第306题累加数是几乎一样的，只不过此题要将斐波那契数列输出出来。注意一个long型的数如果大于INT_MAX的话，直接push_back进int型的数组时，实际存进数组的会是一个负数。所以在返回ans数组之前要判断一下数组中各值，如果有小于0的数，那么返回｛｝。

代码实现

class Solution {
public:
    vector<int> splitIntoFibonacci(string S) {
        int n = S.size();
        for(int i = 0; i < 10; i++){
            vector<int> ans;
            if(i + 1 > n)
                break;
            string s1 = S.substr(0, i + 1);
            if(s1.size() > 1 && s1[0] == '0')
                break;
            long i1 = stol(s1);
            for(int j = i + 1; j < i + 11; j++){
                ans.clear();
                ans.push_back(i1);
                if(j + 1 > n)
                    break;
                string s2 = S.substr(i + 1, j - i);
                if(s2.size() > 1 && s2[0] == '0')
                    break;
                long i2 = stol(s2);
                ans.push_back(i2);
                long i3 = i1 + i2;
                ans.push_back(i3);
                string s3 = to_string(i3);
                if(j + 1 + s3.size() > n)
                    break;
                if(s3 != S.substr(j + 1, s3.size()))
                    continue;
                string all = s1 + s2 + s3;
                long t1 = i1, t2 = i2, t3 = i3;
                while(all.size() < n){
                    t1 = t2;
                    t2 = t3;
                    t3 = t1 + t2;
                    s3 = to_string(t3);
                    ans.push_back(t3);
                    all += s3;
                }
                for(int a : ans)
                    if(a < 0)
                        return {};
                if(all == S)
                    return ans;
            }
        }
        return {};
    }
};