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 <random>
#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<<i<<' '<<val[now]<<endl;
        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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章