后缀字符串(计蒜客:2019 蓝桥杯省赛 B 组模拟赛(一):字典树)

一天蒜头君得到 n个字符串 si​,每个字符串的长度都不超过 10。

蒜头君在想,在这 n 个字符串中,以 si​ 为后缀的字符串有多少个呢?

输入格式

第一行输入一个整数 n。

接下来 n 行,每行输入一个字符串 si​。

输出格式

输出 n 个整数,第 i个整数表示以 si​ 为后缀的字符串的个数。

数据范围

对于 50% 的数据,1<=n <=10^3。

对于 100% 的数据,1<=n <=10^5。

所有的字符串仅由小写字母组成。

样例输入

3
ba
a
aba

样例输出

2
3
1

题意描述:求在这 n 个字符串中,以 si​ 为后缀的字符串有多少个。

分析:其实这道题可以变换一下把后缀换成前缀,怎样实现呢,就是让输入的数组反转一下,然后问题就变为以 si​ 为后缀的字符串有多少个,现在这道题就跟HDU上的统计难题的思路差不多了,算法实现:建立字典树。

代码如下:

#include
#include
#include
char str[100005][11];
struct node//定义结构体
{
	int number;
	node *next[26];
	node()//初始化节点 
	{
		number=0;//个数清零
		memset(next,0,sizeof(next));//指针数组清零
	}
}; 
node *root=NULL;//定义根节点 

void build(char s[],int len)//建立字典树函数
{
	node *p=root;//把根节点赋给p
	int i;
	for(i=0;inext[s[i]-'a']==NULL)//如果目前节点为空,则建立新节点存储这个字符 
		{
			p->next[s[i]-'a']=new node;
		}
		p=p->next[s[i]-'a'];//把p指向p->next[s[i]-'a']
		p->number++;//以当前结点为前缀的单词个数+1 
	}
}
int search(int j,int len)//查找字典树函数
{
	node *p=root;//把根节点赋给p
	int i;
	for(i=0;inext[str[j][i]-'a']==NULL)//走到字典树的叶子节点该单词str还没结束,以该串为前缀的单词数为0
		{
			return 0;
		}
		p=p->next[str[j][i]-'a'];//指向下一个节点
	}
	return p->number; //返回以str为前缀的单词个数
}
int main()
{
	int n,len,i=1,k,j,count;
	char s[11],s1[11];
	root=new node;//生成根节点
	memset(str,0,sizeof(str));
	scanf("%d",&n);
	getchar();
	while(n--)
	{
		
		gets(s1);
		len=strlen(s1);
		for(k=0;k

 

你可能感兴趣的:(后缀字符串(计蒜客:2019 蓝桥杯省赛 B 组模拟赛(一):字典树))