Trie(字典树)非常好用的数据结构,拥有线性时间复杂度。详细请见ACwing~ 。~
来道例题:
前缀统计
给定N个字符串S1,S2…SN,接下来进行M次询问,每次询问给定一个字符串T,求S1~SN中有多少个字符串是T的前缀。
输入字符串的总长度不超过106,仅包含小写字母。
输入格式
第一行输入两个整数N,M。
接下来N行每行输入一个字符串Si。
接下来M行每行一个字符串T用以询问。
输出格式
对于每个询问,输出一个整数表示答案。
每个答案占一行。
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1000010;
char str[N];
int cnt[N],son[N][26],idx;
int n,m;
void insert(){
int p=0;
for(int i=0;i<strlen(str);i++)
{
auto &temp=son[p][str[i]-'a'];
if(!temp) temp=++idx;
p=temp;
}
cnt[p]++;
}
int query()
{
int p=0,res=0;
for(int i=0;i<strlen(str);i++)
{
auto temp=son[p][str[i]-'a'];
if(!temp) break;
else
{
res+=cnt[temp];
p=temp;
}
}
return res;
}
int main(){
//freopen("data.in","r",stdin);
//freopen("data.out","w",stdout);