【ybtoj 高效进阶 2.4】A.前缀统计【trie树】

题目链接：前缀统计

分析

trie树模板题，记板子！

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

int n,m;
int v[1000001];
int a[1000001][27],tot=1;

void in(string s)
{
	int l=s.size(),now=1;
	for(int i=0;i<l;i++)
	{
		if(!a[now][s[i]-'a'])
		{
			tot++;
			a[now][s[i]-'a']=tot;
		}
		now=a[now][s[i]-'a'];
	}
	v[now]++;
}

int find(string s)
{
	int l=s.size(),now=1,ans=0;
	for(int i=0;i<l;i++)
	{
		if(!a[now][s[i]-'a']) break;
		now=a[now][s[i]-'a'];
		ans+=v[now];
	}
	return ans;
}

int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		string s;
		cin>>s;
		in(s);
	}
	for(int i=1;i<=m;i++)
	{
		string s;
		cin>>s;
		cout<<find(s)<<endl;
	}
	return 0;
}