[Swift]字典树-Trie

★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★
➤微信公众号：山青咏芝（let_us_code）
➤博主域名：https://www.zengqiang.org
➤GitHub地址：https://github.com/strengthen/LeetCode
➤原文地址：https://www.cnblogs.com/strengthen/p/11923061.html
➤如果链接不是山青咏芝的博客园地址，则可能是爬取作者的文章。
➤原文已修改更新！强烈建议点击原文地址阅读！支持作者！支持原创！
★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★

Trie树，字典树是一种树形结构，是一种哈希树的变种。典型应用是用于统计，排序和保存大量的字符串（但不仅限于字符串），所以经常被搜索引擎系统用于文本词频统计。它的优点是：利用字符串的公共前缀来减少查询时间，最大限度地减少无谓的字符串比较，查询效率比哈希树高。

例如我们往字典树中插入see、pain、paint三个单词，Trie字典树如下所示：

 参见：[Swift]LeetCode208. 实现 Trie (前缀树) | Implement Trie (Prefix Tree)

class Node {
    var wordEnd = false
    var next : [Character : Node] = [:]
}

class Trie {
    let root = Node()
    /** Initialize your data structure here. */
    init() {
        
    }
    
    /** Inserts a word into the trie. */
    func insert(_ word: String) {
      var currentNode = root
        for c in word {
            if let nextNode = currentNode.next[c] {
                currentNode = nextNode
            } else {
                let nextNode = Node()
                currentNode.next[c] = nextNode
                currentNode = nextNode
            }
        }
        currentNode.wordEnd = true
    }
    
    /** Returns if the word is in the trie. */
    func search(_ word: String) -> Bool {
      var currentNode = root
      for c in word {
          if let node = currentNode.next[c] {
              currentNode = node
          } else {
              return false
          }
      }
        return currentNode.wordEnd
    }
    
    /** Returns if there is any word in the trie that starts with the given prefix. */
    func startsWith(_ prefix: String) -> Bool {
      var currentNode = root
      for c in prefix {
          if let node = currentNode.next[c] {
              currentNode = node
          } else {
              return false
          }
      }
        return true
    }
}

/**
 * Your Trie object will be instantiated and called as such:
 * let obj = Trie()
 * obj.insert(word)
 * let ret_2: Bool = obj.search(word)
 * let ret_3: Bool = obj.startsWith(prefix)
 */