直接套字典树模板就好 判断输入的时候需要注意一下
ac代码
#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <vector>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <utility>
#define pi 3.1415926535898
#define ll long long
#define lson rt<<1
#define rson rt<<1|1
#define eps 1e-6
#define ms(a,b) memset(a,b,sizeof(a))
#define legal(a,b) a&b
#define print1 printf("111\n")
using namespace std;
const int maxn = 5e5+10;
const int inf = 0x3f3f3f3f;
const ll llinf =0x3f3f3f3f3f3f3f3f;
const int mod = 2333;
char a[30];
char b[30];
struct Trie
{
Trie *nex[26];
int flag;
Trie()
{
flag=1;
memset(nex,NULL,sizeof(nex));
}
}*root;
void inserts(char *str)
{
int len=strlen(str);
Trie *p=root,*q;
for(int i=0;i<len;i++)
{
int id=str[i]-'a';
if(p->nex[id]==NULL)
{
q=new Trie();
p->nex[id]=q;
p=p->nex[id];
}else
{
p=p->nex[id];
++(p->flag);
}
}
}
int query(char *str)
{
int len=strlen(str);
Trie *p=root;
for(int i=0;i<len;i++)
{
int id=str[i]-'a';
p=p->nex[id];
if(p==NULL)return 0;
}
return p->flag;
}
void free(Trie *t)
{
if(t==NULL)return;
for(int i=0;i<26;i++)
{
if(t->nex[i])free(t->nex[i]);
}
delete(t);
}
int main()
{
root=new Trie();
while(gets(a)&&a[0]!='\0')
{
inserts(a);
}
//printf("\n");
while(scanf("%s",b)!=EOF)
{
printf("%d\n",query(b));
}
free(root);
return 0;
}