Leetcode 306. 累加数 C++

Leetcode 306. 累加数

题目

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

一个有效的累加序列必须至少包含 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

题解

此问题重点就是确定前两个数，确定完这两个数后，后面的数就确定了，只需要进行判断即可。详细过程见代码

代码

	bool ans = false;
    void dfs(int num,int now,string& str,int len){
        if(now == 0){		//确定第一个数
            for(int i=now; i<len/2; i++){
                if(i!=now && num==0)    break; 
                num = num*10 + (str[i]-'0');
                dfs(num,i+1,str,len);
                if(ans) break;
            }
        }else{
            long x = 0;
            int i=now;
            int old;
            while(i<=len*2/3){		//确定第二个数
                old = num;
                if(i!=now && x == 0)    break;      //避免前导零
                x = x*10 + (str[i]-'0');
                string flag = to_string(x);
                i++;
                int newi = i;
                while(newi < len){		//根据已知的两个数确定第三个数，并判断字符串中后面的数是否符合要求
                    long y = 0;
                    for(int k=0; k<flag.length(); k++)
                        y = y*10 + (flag[k]-'0');
                    flag = to_string(old+y);		//确定的新数
                    if(flag == str.substr(newi,flag.length())){
                        old = y;
                        newi += flag.length();
                    }else{
                        newi = 0;
                        break;
                    }
                }
                if(i<len && newi >= len){
                    ans = true;
                    break;
                }    
            }
        }
    }
    bool isAdditiveNumber(string num) {
        if(num.length()<3)  return false;
        dfs(0,0,num,num.length());
        return ans;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/additive-number
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。