Additive number is a string whose digits can form additive sequence.
A valid additive sequence should contain at least three numbers. Except for the first two numbers, each subsequent number in the sequence must be the sum of the preceding two.
Given a string containing only digits
'0'-'9'
, write a function to determine if it's an additive number.
Note: Numbers in the additive sequence cannot have leading zeros, so sequence
1, 2, 03
or
1, 02, 3
is invalid.
Example 1:
Input:"112358"
Output: true
Explanation: The digits can form an additive sequence:1, 1, 2, 3, 5, 8.
1 + 1 = 2, 1 + 2 = 3, 2 + 3 = 5, 3 + 5 = 8
Example 2:
Input:"199100199"
Output: true
Explanation: The additive sequence is:1, 99, 100, 199.
1 + 99 = 100, 99 + 100 = 199
Follow up:
How would you handle overflow for very large input integers?
Credits:
Special thanks to @jeantimex for adding this problem and creating all test cases.
这道题定义了一种加法数，就是至少含有三个数字，除去前两个数外，每个数字都是前面两个数字的和，题目中给了许多例子，也限定了一些不合法的情况，比如两位数以上不能以0开头等等，让我们来判断一个数是否是加法数。目光犀利的童鞋应该一眼就能看出来，这尼玛不就是斐波那契数组么，跟另一道 Split Array into Fibonacci Sequence 简直不要太像啊。只不过那道题要返回一种组合方式，而这道题只是问能否拆成斐波那契数列。
开始我还想是否能用动态规划来解，可是发现不会写状态转移方程，只得作罢。其实这题可用Brute Force的思想来解，我们让第一个数字先从一位开始，第二个数字从一位，两位，往高位开始搜索，前两个数字确定了，相加得到第三位数字，三个数组排列起来形成一个字符串，和原字符串长度相比，如果小于原长度，那么取出上一次计算的第二个和第三个数，当做新一次计算的前两个数，用相同的方法得到第三个数，再加入当前字符串，再和原字符串长度相比，以此类推，直到当前字符串长度不小于原字符串长度，比较两者是否相同，相同返回true，不相同则继续循环。如果所有情况都遍历完了还是没有返回true，则说明不是Additive Number，返回false，参见代码如下：
解法一：
class Solution {
public:
bool isAdditiveNumber(string num) {
for (int i = ; i < num.size(); ++i) {
string s1 = num.substr(, i);
if (s1.size() > && s1[] == '') break;
for (int j = i + ; j < num.size(); ++j) {
string s2 = num.substr(i, j - i);
long d1 = stol(s1), d2 = stol(s2);
if ((s2.size() > && s2[] == '')) break;
long next = d1 + d2;
string nextStr = to_string(next);
if (nextStr != num.substr(j, nextStr.length())) continue; // optimization here
string allStr = s1 + s2 + nextStr;
while (allStr.size() < num.size()) {
d1 = d2;
d2 = next;
next = d1 + d2;
nextStr = to_string(next);
allStr += nextStr;
}
if (allStr == num) return true;
}
}
return false;
}
};
此题还有递归解法，博主最先尝试的是跟那道 Split Array into Fibonacci Sequence 一样的递归方法，虽然这道题不用返回组合方式，但是我们仍可以使用一样的递归来做，只不过这里的结果res是一个布尔型的全局变量，当我们找到一组符合题意的组合了之后，就将结果res设置为true，这样一旦再进入递归函数时，只要res为true了，就直接返回即可。之后的部分就基本相同了，可以参见 Split Array into Fibonacci Sequence 中的讲解，注意稍有不同的地方是，这里拆分出来的数字是可以超过整型int范围的，但是貌似不会超过长整型long的范围，所以我们可以加一个检测str的长度大于19就break，因为long的十进制数长度是19位，参见代码如下：
解法二：
class Solution {
public:
bool isAdditiveNumber(string num) {
bool res = false;
vector<long> out;
helper(num, , out, res);
return res;
}
void helper(string& num, int start, vector<long>& out, bool& res) {
if (res) return;
if (start >= num.size()) {
if (out.size() >= ) res = true;
return;
}
for (int i = start; i < num.size(); ++i) {
string str = num.substr(start, i - start + );
if ((str.size() > && str[] == '') || str.size() > ) break;
long curNum = stol(str), n = out.size();
if (out.size() >= && curNum != out[n - ] + out[n - ]) continue;
out.push_back(curNum);
helper(num, i + , out, res);
out.pop_back();
}
}
};
由于这道题并不需要我们返回具体的组合方式，所以也可以不使用上面的写法，而是不停的用当前的两个数，去拼后面的数字，一旦拼不出来了，直接返回false。跟解法一类似，首先用两个for循环来确定前两个数字，然后将后面的整体提取出来，和当前的两个数字一起调用递归，若递归返回true了，则直接返回true，否则继续遍历。
在递归函数中，还是首先检验是否存在leading zeros，然后将 num1 和 num2 分别转为长整型long，相加后再转回字符串，存入sumStr，这时候看sumStr是否和num相等，是的话直接返回true。否则再来检验，若sumStr的长度大于等于num了，返回false，或者在num中取跟sumStr长度相等的子串，若不等于sumStr，说明无法拼出来，也返回false。若都没返回的话，就再次调用递归，不过这次num要去掉和sumStr长度相等的子串，留下后面的部分，此时带入递归的两个数字要变成 num2 和 sumStr，继续跟后面的比较，参见代码如下：
解法三：
class Solution {
public:
bool isAdditiveNumber(string num) {
for (int i = ; i < num.size(); ++i) {
string s1 = num.substr(, i);
if (s1.size() > && s1[] == ) break;
for (int j = i + ; j < num.size(); ++j){
string s2 = num.substr(i, j - i);
if (s2.size() > && s2[] == ) break;
if(helper(num.substr(j), s1, s2)) return true;
}
}
return false;
}
bool helper(string num, string num1, string num2){
if ((num1.size() > && num1[] == '') || (num2.size() > && num2[] == '')) return false;
string sumStr = to_string(stol(num1) + stol(num2));
if (sumStr == num) return true;
if (sumStr.size() >= num.size() || num.substr(, sumStr.size()) != sumStr) return false;
return helper(num.substr(sumStr.size()), num2, sumStr);
}
};
题目中有个 follow up 说是让我们handle超大整数的溢出情况，那么我们想，上面的解法中哪块可能溢出啊，当然只有将子串转为长整型long的时候，假如此时字符串特别长，甚至超出了长整型的范围，那么我们就不能用stol了，此时就不能转换了，那么我们只能强行将两个字符串相加了，这里用的又是另外一道题目 Add Strings 的知识点了，这样我们就完美的避开了可能溢出的情况了，参见代码如下：
解法四：
// Follow up, handle overflow for very large input integers.
class Solution {
public:
bool isAdditiveNumber(string num) {
for (int i = ; i < num.size(); ++i) {
string s1 = num.substr(, i);
if (s1.size() > && s1[] == ) break;
for (int j = i + ; j < num.size(); ++j){
string s2 = num.substr(i, j - i);
if (s2.size() > && s2[] == ) break;
if(helper(num.substr(j), s1, s2)) return true;
}
}
return false;
}
bool helper(string num, string num1, string num2){
if ((num1.size() > && num1[] == '') || (num2.size() > && num2[] == '')) return false;
string sumStr = add(num1, num2);
if (sumStr == num) return true;
if (sumStr.size() >= num.size() || num.substr(, sumStr.size()) != sumStr) return false;
return helper(num.substr(sumStr.size()), num2, sumStr);
}
string add(string num1, string num2) {
string res;
int i = (int)num1.size() - , j = (int)num2.size() - , carry = ;
while (i >= || j >= ) {
int val1 = (i >= ) ? (num1[i--] - '') : ;
int val2 = (j >= ) ? (num2[j--] - '') : ;
int sum = val1 + val2 + carry;
res.push_back(sum % + '');
carry = sum / ;
}
if (carry) res.push_back(carry + '');
reverse(res.begin(), res.end());
return res;
}
};
类似题目：
Add Strings
Split Array into Fibonacci Sequence
参考资料：
https://leetcode.com/problems/additive-number/
https://leetcode.com/problems/additive-number/discuss/75567/Java-Recursive-and-Iterative-Solutions
https://leetcode.com/problems/additive-number/discuss/75704/My-Simple-C%2B%2B-Non-recursion-Solution
https://leetcode.com/problems/additive-number/discuss/75576/0ms-concise-C%2B%2B-solution-(perfectly-handles-the-follow-up-and-leading-0s)
LeetCode All in One 题目讲解汇总(持续更新中...)
