1. 程式人生 > 實用技巧 >leetcode91 解碼方法(Medium)

leetcode91 解碼方法(Medium)

題目來源:leetcode91 解碼方法

題目描述:

一條包含字母 A-Z 的訊息通過以下方式進行了編碼:

'A' -> 1
'B' -> 2
...
'Z' -> 26
給定一個只包含數字的非空字串,請計算解碼方法的總數。

示例 1:

輸入: "12"
輸出: 2
解釋: 它可以解碼為 "AB"(1 2)或者 "L"(12)。

示例 2:

輸入: "226"
輸出: 3
解釋: 它可以解碼為 "BZ" (2 26), "VF" (22 6), 或者 "BBF" (2 2 6) 。

解題思路:

動態規劃。dp[i]為0~i的譯碼總數。
①若s[i]=0,若s[i-1]=1或者2,dp[i]=dp[i-2],否則出錯,返回false
②若s[i-1]=1,那麼dp[i]可以有兩種情況,s[i]和s[i-1]分開譯碼或者不分開,dp[i]=dp[i-1]+dp[i-2]
③若s[i-2]=2,若s[i]在1~6直接,則可以合併譯碼dp[i]=dp[i-1]+dp[i-2],否則只能分開譯碼 dp[i]=dp[i-1]
dp[i]只和前兩個變數有關係,可以用兩個變數代替。

class Solution {
public:
    int numDecodings(string s) {
        int n=s.size();
        if(n==0) return 0;
        if(s[0]=='0') return 0;
        int pre=1,cur=1;
        for(int i=1;i<n;i++){
           int temp=cur;
           if(s[i]=='0'){
               if (s[i-1]=='1'||s[i-1]=='2')
                cur=pre;
               else return 0;
           }
           else if(s[i-1]=='1'||(s[i-1]=='2'&&s[i]>='1'&&s[i]<='6'))
            cur=cur+pre;
           pre=temp;
        }
        return cur;
    }
};