刷题之回溯法：累加数问题

问题描述：

累加数是一个字符串，组成它的数字可以形成累加序列。

一个有效的累加序列必须至少包含 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
进阶:
你如何处理一个溢出的过大的整数输入?

解决方案

class Solution{
    public:
    //两个大数的加法
    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){//逐位相加，记录大于10的余数
            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;
    }
    //核心操作，回溯发，left为剩余的字符串，level为参与计算的数字个数，ans用于记录已经参与计算的字符串
    bool backtrack(string& left, int level, vector<string>& ans){
        //回溯的退出条件，当剩余的字符串为空的时候，需要判断参与计算的数字个数，如果大于3返回true，否则，说明字符串的长度不足3返回false
        if(left.empty()){
            if(level>=3){
                return true;
            }else{
                return false;
            }
        }
        //从0开始遍历
        for(int i =0;i<left.size();i++){
            //num取从索引0开始到i+1的数字，刚开始是一位，后面回溯过程中可能是几位
            string num = left.substr(0,i+1);
            if(level<2||num==add(ans[level-1],ans[level-2])){//如果参与计算的数字的个数不够两个，或者num取得的数字等于前面两个数字的和
                ans.push_back(num);//将数字存入向量
                string substr;//取怒num数字后面的字符串
                if(i+1<left.size()){
                    substr = left.substr(i+1);
                }
                if(backtrack(substr,level+1,ans)){//进入下一层
                    return true;
                }
                ans.pop_back();//回溯，返回说明加入vector的数字不合适返回上一层
            }

            if(left[0] == '0'){//如果开头数字为0，跳过本次循环，进入下一层循环
                break;
            }
        }
        return false;
    }

    bool isAdditiveNumber(string num){
        vector<string> ans;
        return backtrack(num,0,ans);
    }
};

两个大整数相加的操作,以字符串的形式进行

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+=was (t1+t2+carry)%10+'0';
carry = (t1+t2+carry)>10?1:0;
}
reverse(ans.begin(),ans,end());
return ans;
}