现在的位置: 首页 > 综合 > 正文

LeetCode Decode Ways

2019年04月17日 ⁄ 综合 ⁄ 共 764字 ⁄ 字号 评论关闭

  有点意思的题目。用动态规划可以O(n)求解出来:a[i]代表子字符串string(0,i)的可能解码方式,a[i] = {a[i-1] or a[i-1]+a[i-2]}.

  意思是如果string(i)不为0,至少a[i] == a[i-1],即一种解码方法是string{0,.....(i-1)}+string(i); 

  然后如果string{i-1,i}是合法的(注意合法概念,比如11,12,20,但04就不合法),那么a[i] = a[i-1]+a[i-2],即还有一种解码方法是string{0,.....(i-2)}+string{i-1,i}

  另外值得注意是如果a[i]求出来为0,那就可以停止运行了,直接返回0,因为这个字符串没法被解码。

#include<iostream>
using namespace std;


class Solution {
public:
	int numDecodings(string s) {
		if (s == "")
			return 0;
		if (s.at(0) == '0')
			return 0;
		int len = s.size();
		int* a = new int[len];

		a[0] = 1;
		for (int i = 1; i < len; i++)
		{
			a[i] = 0;

			if (s.at(i) != '0')
				a[i] = a[i - 1];
			
			if (s.at(i - 1) != '0')
			{
				int val = (s.at(i - 1) - '0') * 10 + (s.at(i) - '0');
				if (val <= 26 && val > 0)
				{
					if (i >= 2)
						a[i] += a[i - 2];
					else
						a[i] += 1;
				}
			}

			if (a[i] == 0)
				return 0;
		}

		return a[len-1];
	}
};

int main()
{
	Solution sol;
	string s = "12004";

	cout << sol.numDecodings(s) << endl;
}

抱歉!评论已关闭.