hdu 5384 Danganronpa

链接：http://acm.hdu.edu.cn/showproblem.php?pid=5384

思路：没学自动机时以为是道KMP然后就tle了好几把，AC自动机模板题

 #include<cstdio>
 #include<iostream>
 #include<algorithm>
 #include<math.h>
 #include<string.h>
 #include<vector>
 #include<queue>
 #include<iterator>
 #include<vector>
 #include<set>
 #define dinf 0x3f3f3f3f
 #define LL long long
 using namespace std;
 const int N=;

 int n, m;
 string str[N];
 char s[N];

 struct node
 {
     int num;    //以本节点结尾的单词个数
     node *child[];  //孩子
 } pos[N*];
 node *tree_gen;

 int node_cnt;
 node * create_node()
 {
     pos[node_cnt].num=;
     for(int i=; i<; i++)
         pos[node_cnt].child[i]=;
     return &pos[node_cnt++];
 }

 void insert_tree(char *p)
 {
     node *node_p=tree_gen;  //指向树根
     while(*p!='\0')
     {
         if( node_p->child[*p-'a']== )  //还没有这叉，就要建
         {
             node *new_node=create_node();   //创建新节点
             node_p->child[*p-'a']=new_node; //连接
             node_p=new_node;
         }
         else                        //已有这叉，继续往下
             node_p=node_p->child[*p-'a'];
         if( *(p+)=='\0' )
             node_p->num++;      //以此单词为结尾的
         p++;
     }
 }

 int query(string t)
 {
     int ans=;
     node *node_p=tree_gen;  //指向树根
     int r=;
     while( node_p!= && r<t.size() )
     {
         if(node_p->child[ t[r]-'a' ])
             node_p=node_p->child[ t[r]-'a' ];
         else break;
         r++;
         ans+=node_p->num;
     }
     return ans;
 }

 int cal(string &t)
 {
     int ans=;
     for(int i=; i<t.size(); i++)
         ans+=query(t.substr(i));
     return ans;
 }

 int main()
 {
     int t;
     cin>>t;
     while(t--)
     {
         scanf("%d%d", &n, &m);
         node_cnt=;
         tree_gen=create_node();

         for(int i=; i<n; i++)
             cin>>str[i];
         for(int i=; i<m; i++)
         {
             scanf("%s",s);
             insert_tree(s);
         }
         for(int i=; i<n; i++)
             printf("%d\n", cal(str[i]));
     }
     return ;
 }