题意:
中问题。
题解:
字典树的模板题,我是看这位大佬学习的字典树:
https://www.cnblogs.com/TheRoadToTheGold/p/6290732.html
字典树的最大时间复杂度为O(n),n代表字符串的长度。
这道题主要用来练练手,看看模板是否有问题,这道题我分别写了数组版的和指针版的,神奇的是指针版的在G++会超内存,而C++过了。。。不知道哪里有问题,求大佬讲解一下,不甚感激。
数组版
//数组版
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int MAXN=500000+7;
int trie[MAXN][26];
int vis[MAXN];
int tot=1;
void insert(char *s)
{
int rt=0;
for(int i=0;s[i];i++)
{
int x=s[i]-'a';
if(trie[rt][x]==0)
{
trie[rt][x]=tot++;
}
rt=trie[rt][x];
vis[rt]++;
}
}
int find(char *s)
{
int rt=0;
for(int i=0;s[i];i++)
{
int x=s[i]-'a';
if(trie[rt][x]==0)
return 0;
rt=trie[rt][x];
}
return vis[rt];
}
int main()
{
memset(trie,0,sizeof(trie));
memset(vis,0,sizeof(vis));
char s[15];
while(gets(s)&&s[0]!='\0')
insert(s);
while(~scanf("%s",s))
{
printf("%d\n",find(s));
}
}
指针版
//指针版
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
struct node
{
int cnt;
node *next[26];
}*root;
node *build()
{
node *k=new(node);
k->cnt=0;
memset(k->next,0,sizeof(k->next));
return k;
}
void insert(char *s)
{
node *r=root;
for(int i=0;s[i];i++)//这样的写法也是可以的,因为s[i]=='\0'的时候会跳出。
{
int x=s[i]-'a';
if(r->next[x]==0) r->next[x]=build();
r=r->next[x];
r->cnt++;
}
}
int find(char *s)
{
node *r=root;
for(int i=0;s[i];i++)
{
int x=s[i]-'a';
if(r->next[x]==0) return 0;
r=r->next[x];
}
return r->cnt;
}
int main()
{
root=build();
char s[15];
while(gets(s)&&s[0]!='\0')
{
insert(s);
}
while(~scanf("%s",s))
{
printf("%d\n",find(s));
}
}
//这道题的指针版要用C++才能过。。。。不知道为什么。。
还有一个指针版
//另外一个指针版
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
struct node
{
int cnt;
node *next[26];
node(){
cnt=0;
for(int i=0;i<26;i++)
next[i]=0;
}
}*root;
void insert(char *s)
{
node *r=root;
for(int i=0;s[i];i++)//这样的写法也是可以的,因为s[i]=='\0'的时候会跳出。
{
int x=s[i]-'a';
if(r->next[x]==0) r->next[x]=new(node);
r=r->next[x];
r->cnt++;
}
}
int find(char *s)
{
node *r=root;
for(int i=0;s[i];i++)
{
int x=s[i]-'a';
if(r->next[x]==0) return 0;
r=r->next[x];
}
return r->cnt;
}
int main()
{
root=new(node);
char s[15];
while(gets(s)&&s[0]!='\0')
{
insert(s);
}
while(~scanf("%s",s))
{
printf("%d\n",find(s));
}
}