HihoCoder - 1014 （字典树）

HihoCoder - 1014 （字典树）

题意

  给出一系列单词，再给出一些询问，问询问中每个单词是多少个单词的前缀。

解题思路

  字典树模板题。

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+5;

int trie[maxn][26],sum[maxn],cnt;
char str[15];
void insert()
{
    int len=strlen(str),root=0;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'a';
        if(!trie[root][id]) trie[root][id]=++cnt;
        sum[trie[root][id]]++;
        root=trie[root][id];
    }
}
int search()
{
    int len=strlen(str),root=0;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'a';
        if(!trie[root][id]) return 0;
        root=trie[root][id];
    }
    return sum[root];
}
int main()
{
    int n,m;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%s",str);
        insert();
    }
    scanf("%d",&m);
    for(int i=0;i<m;i++)
    {
        scanf("%s",str);
        printf("%d\n",search());
    }
    return 0;
}