public string LongestWord(string[] words)
{
// 首先构建前缀树的根节点
PrefixTreeNode root = new PrefixTreeNode();
root.word = "_";
// 利用单词数组构建前缀树
foreach (string word in words)
{
root.insert(word);
}
string[] result = new String[] { "z" };
// 在前缀树中搜索
dfs(root, result);
return(result[0]);
}
/**
* 实现向前缀树中插入一个单词
*/
public void insert(String s)
{
char[] chs = s.ToCharArray();
PrefixTreeNode curNode = this;
// 遍历当前单词中的每一个字符
for (int i = 0; i < chs.Length; i++)
{
// 判断当前字符对应的节点是否在curNode的索引数组中，不存在则加入该字符对应节点
int index = chs[i] - 'a';
if (curNode.links[index] == null)
{
curNode.links[index] = new PrefixTreeNode();
}
// 判断下一个字符
curNode = curNode.links[index];
}
// 当前节点对应的单词为s
curNode.word = s;
}
void dfs(PrefixTreeNode node, String[] result)
{
// 如果一个节点包含的单词长度为0则说明原先单词数组中没有单词与该节点对应，直接返回
if (node == null || node.word.Length == 0)
{
return;
}
foreach (PrefixTreeNode child in node.links)
{
if (child != null)
{
dfs(child, result);
// 寻找最长单词并且长度相等的情况下返回最小的，因此在这里进行判断
bool valid = child.word.Length > result[0].Length ||
(child.word.Length == result[0].Length &&
child.word.CompareTo(result[0]) < 0);
if (valid)
{
result[0] = child.word;
}
}
}
}
