142. 前缀统计 AcWing

原题链接

Trie的基本运用

错误思路：

       将要查找前缀的字符串构建字典树,这样的结果是每个字符串都要重新构建一次树,并且我们需要预先保存要匹配前缀的单词,但题目单词数目没有讲明,所以我们必须将建树的字符串互换.(这样建树会导致MLE)

正解思路：

       将前缀建树,如果达到一个结点有单词就+1,如果没有单词就跳出

易错：

      我们需要再当前结点的下一节点处看是否有单词,否则会少计数

      可能会有重复单词

#include <iostream>
#include <string>
#include <cstring>
using namespace std;
const int N = 1e6+10;
int son[N][28],idx,cnt[N*28];
void insert(string s)
{
    int p = 0;
    for(int i=0;i<s.size();i++){
        int u = s[i]-'a';
        if(!son[p][u]) son[p][u] = ++idx;
        p = son[p][u];
    }
    cnt[p]++;
}
int query(string t)
{
    int p = 0,ans = 0;
    for(int i=0;i<t.size();i++){
        int u = t[i]-'a';
        if(!son[p][u]) break;
        if(cnt[p]) ans+=cnt[p];
        p = son[p][u];
    }
    return ans;
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        string s;
        cin>>s;
        insert(s);
    }
    for(int i=1;i<=m;i++){
        string s;
        cin>>s;
        int ans = query(s);
        printf("%d\n",ans);
    }

    return 0;
}