洛谷 P8306 【模板】字典树

合集 - 3.26~4.13 蓝桥杯省赛备赛(28)

1.P8687 [蓝桥杯 2019 省 A] 糖果2024-03-252.洛谷 P1115 最大子段和2024-03-253.写模板， 并查集。2024-03-254.洛谷 P1656 炸铁路2024-03-265.写模板，树状数组。2024-03-266.洛谷 P3374 【模板】树状数组 12024-03-267.洛谷 P3368 【模板】树状数组 22024-03-268.洛谷 P9237 [蓝桥杯 2023 省 A] 像素放置2024-03-269.洛谷 P5937 [CEOI1999] Parity Game2024-03-27

10.洛谷 P8306 【模板】字典树2024-03-27

11.写模板，kmp。2024-03-2712.写模板， LCA。2024-03-2813.洛谷 P1967 [NOIP2013 提高组] 货车运输2024-03-2814.写模板，线段树2024-03-2815.蓝桥杯-百亿富翁2024-03-2916.洛谷 P3372 【模板】线段树 12024-03-2917.模拟比赛-14届研究生组C++省赛2024-04-0118.写模板， 线性筛2024-04-0119.写模板-质数的简单应用2024-04-0220.洛谷 P1196 [NOI2002] 银河英雄传说2024-04-0421.洛谷 P1004 [NOIP2000 提高组] 方格取数2024-04-0422.蓝桥杯 计算方程2024-04-0423.洛谷 P1006 [NOIP2008 提高组] 传纸条2024-04-0524.蓝桥杯-算法赛第9场强者：贝贝的2.02024-04-0825.P1002 [NOIP2002 普及组] 过河卒2024-04-0826.蓝桥杯，推导部分和2024-04-0827.牛客，小红不想做完全背包2024-04-0928.蓝桥杯-岛屿个数2024-04-10

收起

写模板：
1 确定树的节点指针数量
2 确定起始字符
3 实现插入方法
4 根据题目编写求解方法，或者添加计数元素到节点中

struct Node{
    array<int, 100> next{};
    int cnt = 0;
};

class Trie{
public:
    Trie(char start): start_(start), root_(0){
        trie_.resize(1);
    }

    void insert(const string& s){
        int cur = root_;
        for (const auto& x : s){
            int p = x - start_;
            if (trie_[cur].next[p] == 0){
                trie_.emplace_back();
                trie_[cur].next[p] = trie_.size() - 1;
            }
            cur = trie_[cur].next[p];
            trie_[cur].cnt ++;
        }
    }

    int get(const string& t){
        int cur = root_;
        for (const auto& x : t){
            int p = x - start_;
            if (trie_[cur].next[p] == 0){
                return 0;
            }
            cur = trie_[cur].next[p];
        }
        return trie_[cur].cnt;
    }

private:
    vector<Node> trie_;
    char start_;
    int root_;
};


void solve(){
    int n, q;
    cin >> n >> q;

    Trie trie('0');
    for (int i = 0; i < n; ++i){
        string s;
        cin >> s;
        trie.insert(s);
    }

    while (q --){
        string t;
        cin >> t;
        cout << trie.get(t) << '\n';
    }
}