[ybtoj]前缀统计

SSL_lyw

于 2021-07-11 19:28:51 发布

阅读量161

点赞数 1

分类专栏： YbtOJ

本文链接：https://blog.youkuaiyun.com/SSL_lyw/article/details/118658469

版权

字符串匹配字典前缀统计计数 C++代码

关键词由优快云通过智能技术生成

YbtOJ 专栏收录该内容

24 篇文章

订阅专栏

前缀统计

题目描述

在这里插入图片描述

样例输入

3 2
ab
bc
abc
abc
efg

样例输出

2
0

数据范围

在这里插入图片描述

解题思路

字典数模版

Code

#include<iostream>
#include<cstdio>
using namespace std;
int n,m,sum,c[1000005],t[1000005][30];
string s; 
void insert()
{
	int o=0,len=s.size();
	for(int i=0;i<len;i++)
	{
		int ch=s[i]-'a';
		if(!t[o][ch])t[o][ch]=++sum;
		o=t[o][ch];
	}
	c[o]++;
	return;
}
int get()
{
	int o=0,ans=0,len=s.size();
	for(int i=0;i<len;i++)
	{
		int ch=s[i]-'a';
		o=t[o][ch];
		if(!o)return ans;
		ans+=c[o];
	}
	return ans;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		cin>>s;
		insert();
	}
	for(int i=1;i<=m;i++)
	{
		cin>>s;
		printf("%d\n",get());
	}
	return 0;
}