hdu 2896 多模式串匹配2

hdu 2896 多模式串匹配2

也是一道AC自动机解决多模式串匹配的问题，注意的是用一个visited记录id的出现，以及病毒的数组需要排序，当病毒数超过3时可以结束匹配。
代码如下：

  1 #include  < stdio.h >
  2 #include  < queue >
  3 #include  < vector >
  4 #include  < algorithm >
  5 using   namespace  std;
  6
  7 struct  TrieNode 
  8 {
  9    int flag;
 10    int fail;
 11    int id;
 12    int next[128];
 13    void init()
 14    {
 15        flag = 0;
 16        id = 0;
 17        fail = -1;
 18        for (int i = 0; i < 128; ++i)
 19        {
 20            next[i] = 0;
 21        }
 22    }
 23} ;
 24 const   int  N  =   100000 ;
 25 TrieNode trieTree[N];
 26 bool  visited[ 550 ];
 27 int  cnt  =   0 ;
 28 int  lenB;
 29 int  lenW;
 30
 31 void  insert( const   char *  _str, int  _id)
 32 {
 33    int rt = 0;
 34    while(*_str)
 35    {
 36        int t = *_str;
 37        if (trieTree[rt].next[t] == 0)
 38        {
 39            trieTree[++cnt].init();
 40            trieTree[rt].next[t] = cnt;
 41        }
 42        rt = trieTree[rt].next[t];
 43        ++_str;
 44    }
 45    trieTree[rt].flag++;
 46    trieTree[rt].id = _id;
 47}
 48
 49 void  buildAC()
 50 {
 51    queue<int> Que;
 52    Que.push(0);
 53    int now;
 54    while(!Que.empty())
 55    {
 56        now = Que.front();
 57        Que.pop();
 58        for (int i = 0; i < 128; ++i)
 59        {
 60            if(trieTree[now].next[i])
 61            {
 62                int p = trieTree[now].fail;
 63                int q = trieTree[now].next[i];
 64                while(p != -1 && trieTree[p].next[i] == 0)
 65                    p = trieTree[p].fail;
 66                if (p == -1)
 67                {
 68                    trieTree[q].fail = 0;
 69                }
 70                else
 71                {
 72                    trieTree[q].fail = trieTree[p].next[i];
 73                }
 74                Que.push(q);
 75            }
 76        }
 77    }
 78}
 79
 80 char  buf[ 20000 ];
 81
 82 vector < int >  Match( const   char *  _str)
 83 {
 84    vector<int> ret;
 85    int rt = 0;
 86    int p;
 87    while(*_str)
 88    {
 89        int t = *_str;
 90        if (trieTree[rt].next[t])
 91        {
 92            rt = trieTree[rt].next[t];
 93        }
 94        else
 95        {
 96            p = trieTree[rt].fail;
 97            while(p != -1 && trieTree[p].next[t] == 0)
 98                p = trieTree[p].fail;
 99            if (p == -1)
100            {
101                rt = 0;
102            }
103            else
104            {
105                rt = trieTree[p].next[t];
106            }
107        }
108        p = rt;
109        while(p != 0 && trieTree[p].flag)
110        {
111            if (trieTree[p].flag && !visited[trieTree[p].id])
112            {
113                visited[trieTree[p].id] = true;
114                ret.push_back(trieTree[p].id);
115                if (ret.size() == 3)
116                {
117                    return ret;
118                }
119            }
120            p = trieTree[p].fail;
121        }
122        ++_str;
123    }
124
125    return ret;
126}
127
128 void  Test()
129 {
130    cnt = 0;
131    trieTree[cnt].init();
132    for (int i = 1; i <= lenB; ++i)
133    {
134        scanf("%s",buf);
135        insert(buf,i);
136    }
137    buildAC();
138    scanf("%d",&lenW);
139    int total = 0;
140    for (int i = 0; i < lenW; ++i)
141    {
142        scanf("%s",buf);
143        memset(visited,0,sizeof(visited));
144        vector<int> ret = Match(buf);
145        if (ret.size())
146        {
147            ++total;
148            sort(ret.begin(),ret.end());
149            printf("web %d:",i+1);
150            for (int j = 0; j < ret.size(); ++j)
151            {
152                printf(" %d",ret[j]);
153            }
154            printf("\n");
155        }
156    }
157    printf("total: %d\n",total);
158}
159
160 int  main()
161 {
162    //freopen("data.txt","r",stdin);
163    while(scanf("%d",&lenB) != EOF)
164    {
165        Test();
166    }
167    return 0;
168}