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

论坛 期权论坛 编程之家     
选择匿名的用户   2021-5-23 11:05   11   0

一天蒜头君得到 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<stdio.h>
#include<string.h>
#include<ctype.h>
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;i<len;i++)//循环遍历字符数组s
 {
  if(p->next[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;i<len;i++)//循环遍历字符数组str
 {
  if(p->next[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<len;k++)
   s[k]=s1[len-1-k];//反转字符串,使其把问题由后缀变为前缀 
  s[len]='\0';
  strcpy(str[i],s);
  i++;
  len=strlen(s);
  build(s,len);//建立字典树函数
 }
 j=1;
 while(j<i)
 {
  len=strlen(str[j]);
  count=search(j,len);//查找字典树函数
  printf("%d\n",count);
  j++;
 }
 return 0;
} 

分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:3875789
帖子:775174
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP