leetcode 2227 — 加密解密字符串

一、题目描述

二、算法

这道题使用逆向分析比较简单，我们这里主要是学习下字典树的使用。

1、字典树

字典树（trie），其实就是一个有限状态自动机。一个由 a、an、the 三个单词构成的字典树如图：

通常，我们可以使用状态转移矩阵来保存转换过程：

class Trie {
	static const int MAX_STATUS_NUM = 100000;
	int statusNum = 0;
	int fsm[MAX_STATUS_NUM][26] = { 0 };
	int accept[MAX_STATUS_NUM] = { 0 };

public:
	void insert(const char* s, int len) {
		int status = 0;
		for (int i = 0; i < len; ++i) {
			int index = s[i] - 'a';

			if (fsm[status][index] == 0) {
				fsm[status][index] = ++statusNum;
			}

			status = fsm[status][index];
		}

		accept[status] = true;
	}

	bool find(const char* s, int len) {
		int status = 0;
		for (int i = 0; i < len; ++i) {
			int index = s[i] - 'a';

			if (fsm[status][index] == 0) {
				return false;
			}

			status = fsm[status][index];
		}

		return accept[status];
	}
};

2、实现

这里加密的过程就不说了。解密需要保存所有密文对应的明文。如果节约空间，首选应该是 unordered_multimap。不过，使用这个数据结构在查找时需要耗费一定的时间，算法很容易超时。考虑到每个密文都由两个字符组成，我们使用 26*26 的数组来保存。另外，由于解密是逐字符解密。我们选择使用递归来求解明文字符串个数：

class Encrypter {
	...

	int find(char* s, int l, int start = 0) {
		if (l == 0) {
			return accept[start] ? 1 : 0;
		}
		else if ((l & 0x1) == 1) {
			return 0;
		}

		int ret = 0;
		for (auto& ch : vk[s[0] - 'a'][s[1] - 'a']) {
			int c = ch - 'a';
			if (fsm[start][c] != 0) {
				ret += find(s + 2, l - 2, fsm[start][c]);
			}
		}

		return ret;
	}

	vector<vector<vector<char>>> vk;
public:
	Encrypter(vector<char>& keys, vector<string>& values, vector<string>& dictionary) :
		vk(26, vector<vector<char>>(26))
	{
		for (auto &word : dictionary) {
			insert((char*)word.c_str(), word.length());
		}

		for (int i = 0; i < keys.size(); ++i) {
			vk[values[i][0] - 'a'][values[i][1] - 'a'].push_back(keys[i]);
		}
	}

	int decrypt(string word2) {
		return find((char*)word2.c_str(), word2.length());
	}
};