842. 将数组拆分成斐波那契序列 - 力扣(LeetCode)
标准的枚举题型了,由于存在不能拆分的情况,所以应该使用带返回值的回溯,需要枚举前两个数字,然后根据前两个数字依次检查后面是否存在合法的序列,如果有,则返回true
,否则返回false
。
class Solution {
public:
vector<int> res;
bool dfs(string &s, int idx){//idx表示当前下标
if(idx == s.size()) return res.size() > 2;//回溯出口即用尽s中的字符
long cur = 0;//记录数值
for(int i = idx; i < s.size(); ++i){
cur = 10 * cur + s[i] - '0';//每次循环增加一位数
if(cur > INT_MAX) break;//非int直接break
if(res.size() >= 2 && cur - res[res.size()-1] > res[res.size()-2]) break;//再往后加一位cur只会更大
if(res.size() <= 1 || cur - res[res.size()-1] == res[res.size()-2]){//减法防止溢出
res.push_back(cur);//做出选择
if(dfs(s, i+1)) return true;//只需要找到一种组合即可
res.pop_back();//撤销选择
if(cur == 0) break;//当前是0,就不用往cur后面再追加数字了
}
}
return false;
}
vector<int> splitIntoFibonacci(string &s) {
dfs(s, 0);
return res;
}
};