acwing 字典树/前缀树(Trie)

原创已于 2024-01-08 13:26:41 修改 · 462 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#算法

于 2024-01-08 11:44:43 首次发布

文章介绍了如何使用Trie树实现字符串统计（ACWing835）和查找最大异或对（ACWing143），以及在LeetCode问题中构建节点型前缀树的实例。

AcWing 835. Trie字符串统计

在这里插入图片描述

#include <iostream>
#include <string>
#include <unordered_map>

using namespace std;

const int maxn = 1e5 + 10;
int trie[maxn][26], count[maxn];
int index = 0;


void insert(const string& str)
{
    int cur = 0;
    for(int i = 0; i < str.size(); ++i)
    {
        int col_idx = str[i] - 'a';
        // 当前字符在字典树路径中未出现，插入并指定序号
        if(trie[cur][col_idx] == 0)
            trie[cur][col_idx] = ++index;
        // 沿着树中路径向下查询
        cur = trie[cur][col_idx];
    }
    // 当前路径结尾处标记累加已出现次数
    count[cur]++;
}

int query(const string& str)
{
    int cur = 0;
    for(int i = 0; i < str.size(); ++i)
    {
        int col_idx = str[i] - 'a';
        if(trie[cur][col_idx] == 0)
            return 0;
        cur = trie[cur][col_idx];
    }
    return count[cur];
}

int main()
{
    int N;
    cin >> N;
    char op;
    string str;
    for(int i = 0; i < N; ++i)
    {
        cin >> op >> str;
        if(op == 'I')
            insert(str);
        else
            cout << query(str) << endl;
    }   
    
    return 0;    
}

AcWing 143. 最大异或对

在这里插入图片描述

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1e5 + 10, M = 31e5 + 10;
int n = 0, index = 0;
int nums[N], trie[M][2];
// int re[M];

// 构建树
void insert(int num)
{
    int cur = 0;
    for(int i = 31; i >= 0; --i)
    {
        int col_idx = (num >> i) & 1;
        if(trie[cur][col_idx] == 0)
            trie[cur][col_idx] = ++index;
        cur = trie[cur][col_idx];
    }
    // re[cur] = num;
}

// 查询树中与当前元素异或值最大的节点
// 根据对应比特位上是否相异确定查询路径
int query(int num)
{
    int ret = 0, cur = 0;
    for(int i = 31; i >=0; --i)
    {
        int col_idx = (num >> i) & 1;
        if(trie[cur][!col_idx] != 0)
        {
            ret += 1 << i;
            cur = trie[cur][!col_idx];
        }
        else
            cur = trie[cur][col_idx];
    }
    // return num ^ re[cur];
    return ret;
}

int main()
{
    cin >> n;
    for(int i = 0; i < n; ++i)
    {
        cin >> nums[i];
        insert(nums[i]);
    }
    int ret = 0;
    for(int i = 0; i < n; ++i)
        ret = max(ret, query(nums[i]));
    cout << ret;
    return 0;
}

lc上前缀树实现（节点型）

208. 实现 Trie (前缀树)
在这里插入图片描述

class Trie {
private:
    vector<Trie*> next;
    bool isEnd;
public:
    Trie() 
        :next(26), isEnd(false)
    { }
    
    void insert(string word) {
        Trie *cur = this;
        for(auto &a : word)
        {
            int idx = a - 'a';
            if(cur->next[idx] == nullptr)
                cur->next[idx] = new Trie;
            cur = cur->next[idx];
        }
        cur->isEnd = true;
    }
    
    bool search(string word) {
        Trie *cur = this;
        for(auto &a : word)
        {
            int idx = a - 'a';
            if(cur->next[idx] == nullptr)
                return false;
            cur = cur->next[idx];
        }
        return cur->isEnd;
    }
    
    bool startsWith(string prefix) {
        Trie *cur = this;
        for(auto &a : prefix)
        {
            int idx = a - 'a';
            if(cur->next[idx] == nullptr)
                return false;
            cur = cur->next[idx];
        }
        return true;
    }
};