AcWing 142. 前缀统计（Trie字典树）

AcWing 142. 前缀统计

题目链接：https://www.acwing.com/problem/content/144/

题目大意：

给定N个字符串s1,s2,s3,s4...，接下来进行M次询问，每次询问给出一个字符串T，求s1 -- sn中有多少个字符串是T的前缀。

输入输出：

输入两个整数n， m；然后输入n个字符串。最后输出m个字符串进行询问操作，每次询问输出一个整数res，res表示s1 -- sn中有多少个字符串是当前字符串的前缀。

思路：

这道题很基础的一道Trie字典树的题，考察的是字典树的建树和询问操作，其特点就是上面的加粗字体部分。

这道题我们首先要对输入的n个字符串建立字典树，全部存入trie数组中。然后用一个cnt数组记录每个字符串的末尾位置，这样做的原因是防止有重复的字符串。最后结果res每次加上遍历到当前位置的cnt值即可。

在字典树中，trie数组是关键，通常是建立一个二位trie数组，第一维的下标表示指针的位置，第二维用来表示树中的节点的度，比如本题中每个节点的度大小为26，因为每个节点的孩子节点为26个英文字母，所以trie数组的第二维声明到30就行了。

c++代码：

//AcWing 142. 前缀统计
/**/
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cstring>
#include<cctype>
#include<cmath>
#include<vector>
#include<stack>
#include<queue>
#include<map>
#include<set>

using namespace std;

#define IOS ios::sync_with_stdio(false); cin.tie(0), cout.tie(0);
#define ll long long
#define endl '\n'

typedef pair<char, int> pir;
const int mod = 0x7f7f7f7f;
const int N = 1000010;

int trie[N][30], cnt[N], c;

void insert(string s){
    int q = 0;
    for(int i = 0; i < s.size(); i ++ ){
        int u = s[i] - 'a';        //按照每个字母的ASCLL码来建树
        if(trie[q][u] == 0) trie[q][u] = ++ c;
        q = trie[q][u];            //指针q指向下一个节点
    }
    cnt[q] ++;
}

int query(string s){
    int ans = 0, q = 0;
    for(int i = 0; i < s.size(); i ++ ){
        int u = s[i] - 'a';
        q = trie[q][u];
        if(q == 0) return ans;    //节点为空表示已经到头了，及时return
        ans += cnt[q];            //否则加上该节点的前缀字符串个数
    }
    return ans;
}

int main(void){
    IOS
	
    int n, m; cin >> n >> m;
    while(n -- ){
    	string s; cin >> s;
    	insert(s);
	}
	while(m -- ){
		string s; cin >> s;
		cout << query(s) << endl;
	}

    return 0;
}