Leetcode常用模板-字典树

最新推荐文章于 2025-03-24 15:45:12 发布

原创最新推荐文章于 2025-03-24 15:45:12 发布 · 438 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

LeetCode 专栏收录该内容

2 篇文章

订阅专栏

本文详细介绍了一种高效的数据结构——字典树（Trie），并提供了插入和搜索字符串的具体实现。字典树能有效存储大量字符串，尤其适用于词汇表、IP路由选择等场景，通过递归方式构建节点，实现快速查找。

const int char_size = 26;

struct charTreeNode {
    int check;
    charTreeNode* children[char_size];
    charTreeNode() {
        check = 0;
        for (int i = 0; i < char_size ; i++) {
            children[i] = NULL;
        }
    }
};
struct charTree {
    charTreeNode* root = new charTreeNode();
    void insertNode(string s) {
        int l = s.length();
        charTreeNode* node = root;
        for (int i = 0; i < l; i++) {
            int c = s[i] - 'a';
            if (node -> children[c] == NULL) node -> children[c] = new charTreeNode();
            node = node -> children[c];
        }
        node -> check = 1;
    }
    string searchTree(string s) {
        int l = s.length();
        string ans = "";
        charTreeNode* node = root;
        for (int i = 0; i < l; i++) {
            int c = s[i] - 'a';
            ans += s[i];
            if (node == NULL || node -> children[c] == NULL) return s;
            node = node -> children[c];
            if (node -> check == 1) return ans;
        }
        return s;
    }
};