C/C++编程题刷题：leetcode 842. 将数组拆分成斐波那契序列

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] 也同样被接受。

class Solution {
public:
    vector<int> splitIntoFibonacci(string S) {
        int i=0;
        //i,j,k分别代表第一个、第二个和第三个数字的起始下标
        for(int j=i+1;j<=num.size()-1;j++){
            for(int k=j+1;k<=num.size()-1;k++){
                if(dfs(num,i,j,k)) return true;
            }
        }
        return false;
    }
   
    bool dfs(string& s,int i,int j,int k){
        if((s[i]=='0'&&j-i>1)||(s[j]=='0'&&k-j>1)) 
            return false;
        string a=s.substr(i,j-i);
        string b=s.substr(j,k-j);
        string sum=add(a,b);
        int n=sum.size();
        if(k+n-1>s.size()-1||sum!=s.substr(k,n)) 
            return false;
        if(k+n-1==s.size()-1) 
            return true;
        return dfs(s,j,k,k+n);    
    }

    string add(string& a,string& b){
        int n1=a.size()-1;
        int n2=b.size()-1;
        int carry=0;
        string ans;
        while(n1>=0||n2>=0||carry>0){
            int t1=n1>=0?a[n1--]-'0':0;
            int t2=n2>=0?b[n2--]-'0':0;
            ans+=(t1+t2+carry)%10+'0';
            carry=(t1+t2+carry)>=10?1:0;
        }
        reverse(ans.begin(),ans.end());
        return ans;
    }
};