码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode "Additive Number"

时间:2015-11-19 07:03:34      阅读:128      评论:0      收藏:0      [点我收藏+]

标签:

A natural DFS thought. Several pruning tricks can be applied. Please take care of date type(long long).

class Solution {
    bool _check(string a, string b, string c)
    {    
        int lenc = c.length();
        if (!lenc) return true;
        if(lenc > 1 && c[0] == 0) return false;
        
        int lena = a.length();
        int lenb = b.length();        
        if(lenc < max(lena, lenb) || (lena > 1 && a[0] == 0) || (lenb > 1 && b[0] == 0)) return false;
        
        long long va = stoll(a);
        long long vb = stoll(b);
        
        string ssum = to_string(va + vb);
        int lens = ssum.length();
        if (lens > c.length()) return false;
        string pre = c.substr(0, lens);
        if(pre != ssum) return false;
        
        string rest = c.substr(lens);
        if (rest.length() > 0 && rest[0] == 0) return false;
        
        return _check(b, ssum, rest);
    }
public:
    bool isAdditiveNumber(string num) {
        int n = num.length();
        for(int i = 1; i <= n / 2; i ++) // lena
        {
            string a = num.substr(0, i);
            for(int j = 1; j <= n - 2 * i; j ++) // lenb
            {
                string b = num.substr(i, j);
                string c = num.substr(i + j);
                if (_check(a, b, c))
                {
                    return true;
                }
            }
        }
        return false;
    }
};

LeetCode "Additive Number"

标签:

原文地址:http://www.cnblogs.com/tonix/p/4976455.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!