NYoj 685 查找字符串

描述

小明得到了一张写有奇怪字符串的纸，他想知道一些字符串出现了多少次，但这些字符串太多了，他想找你帮忙，你能帮他吗？输入字符包括所有小写字母、‘@’、‘+’。

 

输入

第一行包含一个整数T（T<=100).表示测试数据组数。
接下来每组数据第一行包含两个整数n，m(n，m<100000),分别表示有n个字符串，小明要问你m次。
接下来n行，每行包含一个字符串，长度不大于15。
接下来m行，每行包含一个字符串，表示小明要问该串出现的次数。

输出

输出每组小明询问数串出现的次数。

样例输入

1
5 3
hello
it@is+so@easy
hello
ibelieveicanac
hello
hello
icannotacit
Giveup

样例输出

3
0
0

题解：水题，就是多判断了“+”，“@”的情况，修改一下模板即可。

代码：

 #include <stdio.h>
 #include <string.h>
 #include <math.h>
 #include <algorithm>
 #include <iostream>
 #include <ctype.h>
 #include <iomanip>
 #include <queue>
 #include <stdlib.h>
 using namespace std;

 struct node
 {
     int count;
     node *next[];
     node(){          //构造函数
         count=;
         memset(next,,sizeof(next));
     }
 };
 node *root;
 int k=;
 void insert(char *a)
 {
     int l=strlen(a);
     node *p=root;
     int i;
     for(i=;i<l;i++)
     {
         if(p->next[a[i]-'+']==)
         {
             p->next[a[i]-'+']=new node;
         }
         // 已存在此前缀
         p=p->next[a[i]-'+'];
         p->count++;
     }
 }
 int find(char *s)
 {
     struct node *p;
     int len=strlen(s);
     if(len==) return ;
     p=root;
     for(int i=;i<len;i++){
         if(p->next[s[i]-'+']!=)
             p=p->next[s[i]-'+'];
         else
             return ;
     }
     return p->count;
 }
 void de(node *p)
 {
     if(p==)
         return ;
     int i;
     for(i=;i<;i++)
     {
         de(p->next[i]);
     }
     delete p;

 }
 int main()
 {
     int t;
     scanf("%d",&t);
     char a[],b[];
     while(t--)
     {
         root = new node;
         int n,m;
         k=;
         scanf("%d%d",&n,&m);
         int i;
         for(i=;i<n;i++)
         {
             scanf("%s",a);
             insert(a);
         }
         for(i=;i<m;i++){
             scanf("%s",b);
             int ans=;
             ans=find(b);
             printf("%d\n",ans);
         }
         de(root);
     }
     return ;
 }