CF514C Watto and Mechanism 蓝 题解

比较好的 Trie 树练手题。

思路

首先，Trie 树有 insert 和 find 两个函数。

insert 使用遍历，在遍历原字符串时，如果未找到这个字符对应的节点，则增添该节点，然后将当前遍历到的节点更新，遍历完整个字符串后标记遍历到的最后一个节点。

void insert (string s) {
	int len=str.size (),root=0;//len 为字符串长度，root 为当前遍历到的节点
	for (int i=0;i<len;i++) {
		int c=str[i]-'a';//当前字符
		if (!trie[root][c])//如果没有该节点
			trie[root][c]=++tot;//增加该节点
		root=trie[root][c];//更新当前遍历点
	}
	flag[root]=1;//标记最后节点
}

find 使用深搜，在递归搜索目标字符串时，如果当前递归到的位置超过字符串总长，则判断是否恰好有一个字符与原字符串不同，且最后一个节点被标记，是则返回 true，否则返回 false。然后判断如果当前已修改超过一次，返回 false。最后判断如果可以更改且改完后符合条件，返回 true。若都不行则返回 false。

bool find (int pos,int len,int root,bool tag) {
	if (pos>len) return !tag&&flag[root];//递归完字符串，判断是否符合条件
	int c=str[pos]-'a';//当前字符
	if (trie[root][c]&&find (pos+1,len,trie[root][c],tag)) return true;
	if (!tag) return false;//如果改过了又要改，返回 false
	for (int j=0;j<3;j++)
		if (c!=j&&trie[root][j]&&find (pos+1,len,trie[root][j],!tag))
			return true;//如何改这一个字符可行，返回 true
	return false;//都不行，返回 false
}

完整代码就不贴了。