trie树 (模板）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/wwwlps/article/details/88683719

#include<bits/stdc++.h>
#define LL long long
using namespace std;
const int maxn=100005;
const int maxz=500005;
int nex[maxn];
string s;

int trie[maxn][26];
int cnt;
int endv[maxn];
void build(string s)
{
    int len=s.size();
    int p=1;
    for(int i=0;i<len;i++)
    {
        if(trie[p][s[i]-'a']==0)
        {
            trie[p][s[i]-'a']=++cnt;
        }
        p=trie[p][s[i]-'a'];
        endv[p]++;
    }
}
int get(string s)
{
    int len=s.size();
    int p=1;
    for(int i=0;i<len;i++)
    {
        if(trie[p][s[i]-'a']==0)
            return 0;
        p=trie[p][s[i]-'a'];
    }
    return endv[p];
}
int main()
{
    ios::sync_with_stdio(false);
    memset(trie,0,sizeof(trie));
    memset(endv,0,sizeof(endv));
    cnt=1;
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>s;
        build(s);
    }
    return 0;
}