C++字典树的写法和应用场景

2023-11-10 02:35:43

字典树简介

字典树（Trie）是一种树形数据结构，通常用于存储字符串，以实现快速检索。它也被称为前缀树，因为它的每个节点都存储着一个字符，并且每个节点的子节点都存储着该字符的所有可能的后续字符。这使得字典树非常适合用于字符串检索，因为只需要从根节点开始，依次比较字符，就可以快速找到目标字符串。

字典树的C++实现

在C++中实现字典树非常简单，我们可以使用一个struct来定义节点，每个节点包含一个字符和一个指向其子节点的数组。然后，我们可以使用一个指向根节点的指针来表示字典树。

struct Node {
  char ch;
  Node* children[26];
  bool is_word;

  Node(char ch) : ch(ch), is_word(false) {
    for (int i = 0; i < 26; ++i) {
      children[i] = nullptr;
    }
  }
};

class Trie {
public:
  Trie() : root(new Node(' ')) {}

  void insert(const string& word) {
    Node* curr = root;
    for (char ch : word) {
      int index = ch - 'a';
      if (curr->children[index] == nullptr) {
        curr->children[index] = new Node(ch);
      }
      curr = curr->children[index];
    }
    curr->is_word = true;
  }

  bool search(const string& word) {
    Node* curr = root;
    for (char ch : word) {
      int index = ch - 'a';
      if (curr->children[index] == nullptr) {
        return false;
      }
      curr = curr->children[index];
    }
    return curr->is_word;
  }

private:
  Node* root;
};