美文网首页动态规划
[leetcode Decode Ways] 掌握动态规划

[leetcode Decode Ways] 掌握动态规划

作者: 书呆子的复仇 | 来源:发表于2015-12-31 17:32 被阅读953次
    原题:

    给定一串数字,求出有多少种编码方式。看完这道题目,第一反应想到的是使用DFS。一个字母最多编码成一个两位数,因此每次遍历的时候要么取一个数字,要么两个数字,每次遍历的时候判断数字是否合法,若能遍历到最后一个数字则编码成功。比如说,我以"101"为例:第一次取"1",然后判断"1"有没有对应的字母,若存在,则截掉第一个数字,剩下"01",继续进行递归,否则直接返回;或者判断"10"有没有对应的字母,若存在,则截掉前面两个数字,剩下"1",继续进行递归,否则直接返回。若最后只剩下一个空字符串,则解码成功。大致就是对一棵二叉树进行遍历一遍。这样的解法在少量数据的情况下还能用。代码大概是这样,在leetcode提交后,毫无悬念得提示超时。

    class Solution {
    public:
        int numDecodings(string s) {
            if (s.empty()) {
                return 0;
            }
            int result = 0;
            dfs(s, &result);
            return result;
        }
        
    private:
        bool isValidLetter(string str) {
            if (str[0] == '0') {
                return false;
            }
            int num;
            if (str.size() == 1) {
                num = str[0] - '0';
            } else {
                num = (str[0] - '0') * 10 + (str[1] - '0');
            }
            return num >= 1 && num <= 26;
        }
        
        void dfs(string str, int *result) {
            
            if (str.size() >= 1) {
                if (isValidLetter(str.substr(0, 1))) {
                    dfs(str.substr(1), result);
                } else {
                    return;
                }
            }
            
            if (str.size() >= 2) {
                if (isValidLetter(str.substr(0, 2))) {
                    dfs(str.substr(2), result);
                } else {
                    return;
                }
            }
            
            if (str.size() == 0) {
                (*result)++;
            }
        }
    };
    

    这道题可以用动态规划来做,dp[i]表示前i个(包括i)数字有多少种解码方式。当i=0时,dp[0] = 0; i = 1呢,dp[1]就等于1吗?未必,可能第一个数字为0,那就无法解码出对应的字母,因此还需要视情况而定。当i=n呢,第n个数字可能单独作为一个字母,也可能与它前一个数字组合成一个字母。因此,我们初步得到状态转移方程:

    (1) dp[0] = 0
    
    (2) dp[1] = 1, if str[1] is valid
        dp[1] = 0, if str[1] is not valid
    
    (3) dp[n] = dp[n - 1] + dp[n - 2]
    

    第三个方程咋看起来好像没有问题,但是第n个数字就一定能保证可以解码成一个合法的字母?同样,第n个数字与它前一个数字的组合就一定保证解码成一个合法的字母?在上述状态转义方程中,我们没有对它进行判断。经过分析后,第三个方程看起来应该是这样的:

    dp[n] = 0
    if str[n] is valid
        dp[n] += dp[n - 1]
    if str[n - 1] combine str[n] is valid
        dp[n] += dp[n - 2]
    

    看起来完美了,但上面的代码还有个小问题。当n = 2时,若第2个字母与第一个字母的组合合法,则dp[2] += dp[0],但是dp[0]我们一开始已经给它赋值为0。因此当n = 2时,即使合法,也相当于没有进行累加。所以,这里要稍加判断,修改后的代码如下:

    dp[n] = 0
    if str[n] is valid
        dp[n] += dp[n - 1]
    if str[n - 1] combine str[n] is valid
        dp[n] += (i - 2 == 0) ? 1 : dp[n - 2]
    

    正确得写出状态转义方程后,实现代码应该不难:

    class Solution {
    public:
        int numDecodings(string s) {
            int dp[s.size() + 1];
            //将数组初始化为0
            memset(dp, 0, sizeof(int) * (s.size() + 1));
            
            //判断第一个数字是否合法,若不合法,直接返回0
            if (s.size() >= 1) {
                if (isValid(s[0])) {
                    dp[1] = 1;
                } else {
                    return 0;
                }
            }
            
            for (int i = 1; i < s.size(); i++) {
                if (isValid(s[i])) {
                    dp[i + 1] += dp[i];
                }
                
                if (isValid(s[i - 1], s[i])) {
                    dp[i + 1] += ((i-1)==0?1:dp[i - 1]);
                }
            }
            return dp[s.size()];
        }
        
    private:
        bool isValid(char c1, char c2) {
            if (c1 == '0') {
                return false;
            }
            int num = (c1 - '0') * 10 + (c2 - '0');
            return num >= 1 && num <= 26;
            
        }
        
        bool isValid(char c) {
            int num = c - '0';
            return num >= 1 && num <= 26;
        }
    };
    

    相关文章

      网友评论

        本文标题:[leetcode Decode Ways] 掌握动态规划

        本文链接:https://www.haomeiwen.com/subject/spaihttx.html