剑指Offer面试题46. 把数字翻译成字符串

Posted 2021-02-18 galaxy-hao

题目

给定一个数字，我们按照如下规则把它翻译为字符串：0 翻译成 “a” ，1 翻译成 “b”，……，11 翻译成 “l”，……，25 翻译成 “z”。一个数字可能有多个翻译。请编程实现一个函数，用来计算一个数字有多少种不同的翻译方法。

示例 1:

输入: 12258
输出: 5
解释: 12258有5种不同的翻译，分别是"bccfi", "bwfi", "bczi", "mcfi"和"mzi"

提示：0 <= num < 231

思路

为了便于处理，将数字转为字符串。dp[i]表示i位置数共有多少种表示，初始dp[0] = 1, 从第1位置开始，如果与前一个位置数组成的值大于等于10并且小于25，则dp[i] = dp[i - 1] + dp[i - 2]（i >= 2），当i = 1，则dp[1] = dp[i - 1] + 1 = 2；否则dp[i] = dp[i - 1]。最后返回dp[size - 1]，size为数字长度。

代码

时间复杂度：O(n)
空间复杂度：O(n)

class Solution {
public:
    int translateNum(int num) {
        string str = to_string(num);
        int size = str.size();
        vector<int> dp(size);
        dp[0] = 1;        
        for (int i = 1; i < size; ++i) {
            if (str.substr(i - 1, 2) >= "10" && str.substr(i - 1, 2) <= "25") dp[i] = dp[i - 1] + (i - 2 >= 0 ? dp[i - 2] : 1);
            else dp[i] = dp[i - 1];
        }
        return dp[size - 1];
    }
};