题目(中等)
累加数 是一个字符串,组成它的数字可以形成累加序列。
一个有效的 累加序列 必须 至少 包含 3 个数。除了最开始的两个数以外,字符串中的其他数都等于它之前两个数相加的和。
给你一个只包含数字 ‘0’-‘9’ 的字符串,编写一个算法来判断给定输入是否是 累加数 。如果是,返回 true ;否则,返回 false 。
说明:累加序列里的数 不会 以 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
提示:
1 <= num.length <= 35
num 仅由数字(0 - 9)组成
进阶:你计划如何处理由过大的整数输入导致的溢出?
思路
暴力枚举第二个数的起止点,如果第二个数的起止点确定,则该累加数就能唯一确定(因为确定了第一个数和第二个数,后面的数可以累加而来),如果确定的累加数和给定累加数相同,则正确,否则为false;
对于每一个暴力选出的第二个数起止点,通过substr确定第一个数和第二个数,通过高精度加法得到第三个数,判断是否在给定序列中,并循环算后续的数,直到结尾处,如果都符合就是true;
中间注意排除掉非零但带前导零的情况。
代码
class Solution {
public:
//默认输入的a和b均为非负整数
string Plus(const string& a, const string& b) {
string ans;
int carry = 0; //进位
for (int i = a.size() - 1, j = b.size() - 1; i >= 0 || j >= 0 || carry != 0; --i, --j) {
int p1 = i >= 0 ? a[i] - '0' : 0, p2 = j >= 0 ? b[j] - '0' : 0;
int k = p1 + p2 + carry;
ans.push_back(k % 10 + '0');
carry = k / 10;
}
reverse(begin(ans), end(ans)); //要进行翻转
return ans;
}
bool istrue(int secondStart, int secondEnd, string num) {
int n = num.size();
int firstStart = 0, firstEnd = secondStart - 1;
string first = num.substr(firstStart, firstEnd - firstStart + 1);
string second = num.substr(secondStart, secondEnd - secondStart + 1);
while(secondEnd <= n - 1) {
string third = Plus(first, second);
int thirdStart = secondEnd + 1;
int thirdEnd = thirdStart + third.size() - 1;
if(thirdEnd >= n) return false;
if(num.substr(thirdStart, third.size()) != third) return false;
if(thirdEnd == n-1) return true;
first = second;
second = third;
secondEnd = thirdEnd;
}
return false;
}
bool isAdditiveNumber(string num) {
int n = num.size();
for(int secondStart = 1; secondStart < n-1; secondStart++) {
if(num[0] == '0' && secondStart != 1) break; //出现非零的0开头数字
for(int secondEnd = secondStart; secondEnd < n-1; secondEnd++) {
if(num[secondStart] == '0' && secondEnd != secondStart) break;
if(istrue(secondStart, secondEnd, num)) return true;
}
}
return false;
}
};