luogu-p2292(AC自动机)

题目描述
标点符号的出现晚于文字的出现,所以以前的语言都是没有标点的。现在你要处理的就是一段没有标点的文章。

一段文章T是由若干小写字母构成。一个单词W也是由若干小写字母构成。一个字典D是若干个单词的集合。我们称一段文章T在某个字典D下是可以被理解的,是指如果文章T可以被分成若干部分,且每一个部分都是字典D中的单词。

例如字典D中包括单词{‘is’, ‘name’, ‘what’, ‘your’},则文章‘whatisyourname’是在字典D下可以被理解的,因为它可以分成4个单词:‘what’, ‘is’, ‘your’, ‘name’,且每个单词都属于字典D,而文章‘whatisyouname’在字典D下不能被理解,但可以在字典D’=D+{‘you’}下被理解。这段文章的一个前缀‘whatis’,也可以在字典D下被理解,而且是在字典D下能够被理解的最长的前缀。

给定一个字典D,你的程序需要判断若干段文章在字典D下是否能够被理解。并给出其在字典D下能够被理解的最长前缀的位置。

输入格式
输入文件第一行是两个正整数n和m,表示字典D中有n个单词,且有m段文章需要被处理。之后的n行每行描述一个单词,再之后的m行每行描述一段文章。

其中1<=n, m<=20,每个单词长度不超过10,每段文章长度不超过1M。

输出格式
对于输入的每一段文章,你需要输出这段文章在字典D可以被理解的最长前缀的位置。

输入输出样例
输入 #1复制
4 3
is
name
what
your
whatisyourname
whatisyouname
whaisyourname
输出 #1复制
14 (整段文章’whatisyourname’都能被理解)
6 (前缀’whatis’能够被理解)
0 (没有任何前缀能够被理解)

题意:给出n个单词,和m段文章,问每段文章中可以由单词组成的最长前缀的长度。

解题思路:使用AC自动机来做,如果我们可以到达这个单词,那么我们也可以到达组成该单词的前面那个单词。这样转移状态即可。
代码:

#pragma GCC optimize(2)
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <string>
#include <vector>
#include <set>
#include <map>
#include <stack>
#include <bitset>
#include <queue>
//#include 
#include <time.h>
using namespace std;
#define int long long
#define ull unsigned long long
#define ls root<<1
#define rs root<<1|1
const int inf=1ll*1<<60;
const int maxn=1e4+10;
const int maxm=0x100010;
const int mod=100000;
char arr[110],str[maxm];
int trie[210][30],fail[210],tot,val[210],flag[210],depth[210];
void add(char *s)
{
    int now=0;
    for(int i=0;s[i]!='\0';i++){
        int x=s[i]-'a';
        if(!trie[now][x])trie[now][x]=++tot,depth[tot]=depth[now]+1;
        now=trie[now][x];
    }
    val[now]++;
}
void getfail()
{
    queue<int> q;
    for(int i=0;i<26;i++){
        if(trie[0][i])q.push(trie[0][i]),flag[trie[0][i]]=1;
    }
    while(!q.empty()){
        int now=q.front();q.pop();
        for(int i=0;i<26;i++){
            if(trie[now][i])fail[trie[now][i]]=trie[fail[now]][i],q.push(trie[now][i]);
            else trie[now][i]=trie[fail[now]][i];
        }
    }
}
int solve(char *s)
{
    int now=0,ans=-1;
    memset(flag,0,sizeof flag);
    for(int i=0;s[i]!='\0';i++){
        int x=s[i]-'a';
        now=trie[now][x];
        //cout<
        for(int j=now;j;j=fail[j]){
            if(val[j] && (flag[i-depth[j]] || i-depth[j]==-1)){
                flag[i]=1;ans=i;break;
            }
        }
    }
    return ans+1;
}
signed main()
{
    int n,m;
    scanf("%lld%lld",&n,&m);
    for(int i=0;i<n;i++){
        scanf("%s",arr);
        add(arr);
    }
    getfail();
    for(int i=0;i<m;i++){
        scanf("%s",str);
        printf("%lld\n",solve(str));
    }
}
//a abcde ef

你可能感兴趣的:(AC自动机)