hdu 2222 - Keywords Search

最新推荐文章于 2021-03-06 10:09:31 发布

小白菜又菜

最新推荐文章于 2021-03-06 10:09:31 发布

阅读量814

点赞数

CC 4.0 BY-SA版权

分类专栏：解题报告字符串

本文链接：https://blog.youkuaiyun.com/mobius_strip/article/details/42062659

解题报告同时被 2 个专栏收录

1404 篇文章

订阅专栏

字符串

121 篇文章

订阅专栏

本文介绍了一个使用AC自动机实现的程序，该程序能够接收一系列单词并统计在一个长文本中这些单词出现的次数，同时确保每个单词仅被计算一次。文章提供了完整的C++代码示例，包括AC自动机的数据结构定义、插入单词、设置失败指针以及查询等功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：给你一些单词构成的词典，统计出现过的单词个数。

分析：字符串，AC自动机。比较裸的AC自动机详细可参照本空间的：AC自动机总结。

说明：注意每个单词只统计一次。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
/* AC_DFA define */
#define nodesize 250010        //节点个数
#define dictsize 26            //符集大小 

typedef struct node1
{
    int    flag;            //值域 
    node1* fail;
    node1* next[dictsize];
}tnode;
tnode  dict[nodesize+1];
tnode* Q[nodesize+1];
int    ID[256]; 

class AC_DFA
{
    private:
        int    size;
        tnode* root;
    public:
        AC_DFA() {
            makeID();
            memset( dict, 0, sizeof( dict ) );
            root=NULL; size=0; root=newnode();
        }
        void makeID() {
            for ( int i = 0 ; i < 26 ; ++ i )
                ID['a'+i] = i;
        }
        void init() {
            memset( dict, 0, sizeof( dict ) );
            root=NULL; size=0; root=newnode();
        }
        tnode* newnode() {
            dict[size].fail = root;
            return &dict[size ++];
        }
        void insert( char* word, int l ) {
            tnode* now = root;
            for ( int i = 0 ; i < l ; ++ i ) {
                if ( !now->next[ID[word[i]]] )
                    now->next[ID[word[i]]] = newnode();
                now = now->next[ID[word[i]]];
            }now->flag ++;
        }
        void setfail() {
            Q[0] = root; root->fail = NULL;
            for ( int move = 0,save = 1 ; move < save ; ++ move ) {
                tnode* now = Q[move];
                for ( int i = 0 ; i < dictsize ; ++ i )
                    if ( now->next[i] ) {
                        tnode* p = now->fail;
                        while ( p && !p->next[i] ) p = p->fail;
                        now->next[i]->fail = p?p->next[i]:root;
                        Q[save ++] = now->next[i];
                    }else now->next[i] = now==root?root:now->fail->next[i];//构建 Trie图 
            }
        }
        int query( char* line, int L ) {//统计字串出现个数，可重复及交叉
            int sum = 0;
            tnode *temp,*now = root;
            for ( int i = 0 ; i < L ; ++ i ) {
                now = now->next[ID[line[i]]];
                temp = now;
                while (temp && temp->flag) {
                	sum += temp->flag;
					temp->flag = 0;
					temp = temp->fail;
				}
            }
            return sum; 
        }
};
/* AC_DFA  end */

char Word[51];
char Line[1000001];

int main()
{
    int T,N;
    while ( ~scanf("%d",&T) ) 
    for ( int t = 1 ; t <= T ; ++ t ) {
        AC_DFA ac;
        scanf("%d",&N);
        for ( int i = 1 ; i <= N ; ++ i ) {
            scanf("%s",Word);
            ac.insert( Word, strlen( Word ) );
        }
        ac.setfail();
        scanf("%s",Line);
        printf("%d\n",ac.query( Line, strlen(Line) ));
    }
    return 0;
}