HDU

来源：互联网发布：贵州大数据战略编辑：程序博客网时间：2024/05/16 12:24

点我看题

题意：给n个模式串，m个主串，问哪些模式串在主串中出现过，输出出现过的模式串的次序。

分析：AC自动机模板题，开一个used数组标记第i个模式串是否在主串中出现过，且每个串的结果直接在query函数中输出。

参考代码：

#include<cstdio>#include<cmath>#include<cstring>#include<set>#include<queue>#include<algorithm>#include<iostream>using namespace std;const int maxn = 5e2+10;const int maxp = 2e2+10;const int maxt = 1e4+10;int n,m;char P[maxp];char T[maxt];int cnt;set<int> ans;struct Aho{    int tt[maxn*maxp][128];    int fail[maxn*maxp];    int endc[maxn*maxp];    int root, L;    int newnode()    {        for( int i = 0; i < 128; i++)            tt[L][i] = -1;        endc[L++] = -1;        return L-1;    }    void init()    {        L = 0;        root = newnode();    }    void insert( char buf[], int id)    {        int len = strlen(buf);        int now = root;        for( int i = 0; i < len; i++)        {            if( tt[now][buf[i]] == -1)                tt[now][buf[i]] = newnode();            now = tt[now][buf[i]];        }        endc[now] = id;    }    void build()    {        queue<int> Q;        fail[root] = root;        for( int i = 0; i < 128; i++)        {            if( tt[root][i] == -1)                tt[root][i] = root;            else            {                fail[tt[root][i]] = root;                Q.push(tt[root][i]);            }        }        while( !Q.empty())        {            int now = Q.front();            Q.pop();            for( int i = 0; i < 128; i++)            {                if( tt[now][i] == -1)                    tt[now][i] = tt[fail[now]][i];                else                {                    fail[tt[now][i]] = tt[fail[now]][i];                    Q.push(tt[now][i]);                }            }        }    }    bool used[510];    bool query( char buf[], int id)    {        int len = strlen(buf);        int now = root;        memset(used,false,sizeof(used));        bool flag = false;        for( int i = 0; i < len; i++)        {            now = tt[now][buf[i]];            int tmp = now;            while( tmp != root)            {                if( endc[tmp] != -1)                {                    used[endc[tmp]] = true;                    flag = true;                }                tmp = fail[tmp];            }        }        if( !flag)            return false;        printf("web %d:",id);        for( int i = 1; i <= n; i++)            if( used[i])                printf(" %d",i);        puts("");        return true;    }};Aho ac;int main(){    while( ~scanf("%d",&n))    {        ac.init();        for( int i = 1; i <= n; i++)        {            scanf("%s",P);            ac.insert(P,i);        }        ac.build();        scanf("%d",&m);        cnt = 0;        for( int i = 1; i <= m; i++)        {            scanf("%s",T);            if( ac.query(T,i))                cnt++;        }        printf("total: %d\n",cnt);    }    return 0;}

阅读全文

0 0