LOJ10053

LOJ10053

字典单词长度这么小，直接写个Trie（真真正正的字典树）
然后类似DP刷表法一样，更新下一状态
如果当前状态能扩展出来，才继续扩展
$O(m*len_{文章}*len_{单词})$

#include<bits/stdc++.h>
#define gt() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++)
#define pt(ch) (Top<1000000?St[Top++]=ch:(fwrite(St,1,1000000,stdout),St[(Top=0)++]=ch))
using namespace std;
int Top;static char St[1000000],buf[1000000],*p1=buf,*p2=buf;
const int maxn=(1<<20)+5,maxl=30;
int n,m,len,f[maxn],ss[maxn];
int read(){
    int ret=0;char ch=gt();
    while(ch<'0'||ch>'9') ch=gt();
    while(ch>='0'&&ch<='9') ret=ret*10+ch-'0',ch=gt();
    return ret;
}
struct Trie{
    int tot,s[205][maxl];bool id[205];
    void insert(){int rot=0;for(int i=1;i<=len;i++) rot=s[rot][ss[i]]=s[rot][ss[i]]?s[rot][ss[i]]:++tot;id[rot]=1;}
    int check(int k){
        int Ans=0;f[0]=k;
        for(int i=0;i<=len;i++){
            if(f[i]!=k) continue;
            Ans=i;
            for(int j=i+1,rot=0;j<=len;j++) if(s[rot][ss[j]]) f[j]=id[rot=s[rot][ss[j]]]?k:f[j];else break;
        }
        return Ans;
    }
}T;
void write(int x){if(x>9) write(x/10);pt(x%10+'0');}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        char ch=gt();len=0;
        while(!islower(ch)) ch=gt();
        while(islower(ch)) ss[++len]=ch-'a',ch=gt();
        T.insert();
    }
    for(int i=1;i<=m;i++){
        char ch=gt();len=0;
        while(!islower(ch)) ch=gt();
        while(islower(ch)) ss[++len]=ch-'a',ch=gt();
        write(T.check(i)),pt('\n');
    }
    fwrite(St,1,Top,stdout);
    return 0;
}