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=100010;

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

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

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

void insert_tree(char *p)
{
    node *node_p=tree_gen;  //指向树根
    while(*p!='')
    {
        if( node_p->child[*p-'a']==0 )  //还没有这叉，就要建
        {
            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+1)=='' )        
            node_p->num++;      //以此单词为结尾的
        p++;
    }
}

int query(string t)
{
    int ans=0;
    node *node_p=tree_gen;  //指向树根
    int r=0;
    while( node_p!=0 && 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=0;
    for(int i=0; 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=0;
        tree_gen=create_node();

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