LINTCODE——Add Operators

来源:互联网 发布:sentinel数据下载 编辑:程序博客网 时间:2024/06/15 18:31

LINTCODE——Add Operators

思路:用回溯法去试,注意两个特殊情况:
一、数字越界,如果大于INT_MAX则溢出,不过给的数据没有这个,我把越界的检查删除了也AC了;
二、数字的有效性,即0987不是有效数字,这种情况要排除,排除方法也很简单,例如已知字符串str =“0987”,把str转换成long long 或者long型得到curNum,再次把curNum,to_string一下,比较两者字符串的长度就OK了;

class Solution {private:    vector<string> ans;public:    /*     * @param num: a string contains only digits 0-9     * @param target: An integer     * @return: return all possibilities     */    vector<string> addOperators(string &num, int target) {        // write your code here        for(int i = 1; i <= num.size(); i++)        {            string curStr = num.substr(0,i);            long long curNum = stol(curStr);            if(to_string(curNum).size() != i)                continue;            dfs(num, target, curStr, i, curNum, curNum, '#');        }        return ans;    }    //curStr为当前字符串,即加了op运算符的字符串    //index为num的索引,判断是否结束递归    //cv为当前字符串的值;    //pv为前一块运算的值    //op为上一个运算符的符号,初试为‘#’    void dfs(const string &num, const int &target, string curStr, int index, long long cv, long long pv, char op)    {        if(index == num.size() && cv == target)        {            ans.push_back(curStr);            return;        }        for(int i = index+1; i <= num.size(); i++)        {            string s = num.substr(index,i-index);            long long curNum = stol(s);            if(to_string(curNum).size() != s.size() )                continue;            dfs(num, target, curStr+'+'+s, i, cv+curNum, curNum, '+');            dfs(num, target, curStr+'-'+s, i, cv-curNum, curNum, '-');            dfs(num, target, curStr+'*'+s, i,(op=='+')?(cv-pv+pv*curNum):((op=='-')?cv+pv-pv*curNum: cv*curNum), curNum*pv, op);        }        return;    }};
原创粉丝点击