jzoj 5795.[2018提高组] 模拟A组&省选 词典 字典树

Description

Input

第一行两个数n,m，表示有n个字符串，m个询问。
接下来n行，每行一个字符串Ti 。
再接下来m行，每行一个字符串Si 。

Output

对于每个询问，输出一个ansi表示答案。

Sample Input

3 2
abcabc
aabc
abbc
aa
ba

Sample Output

1
3

Data Constraint

分析：
直接建一棵字典树（前缀树），每个点记录上一个经过这个点的是哪一个字符串（即是前一个字符串的前缀是当前点），然后这个段的$0$个数相当于这两个字符串的编号差再减$1$，还要考虑上最后面多余的$0$，取一个$max$记录在字典树上即可。

代码：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#define LL long long

const int maxn=5e6+7;

using namespace std;

char s[maxn];
int n,m,k,cnt,cnt1;

struct node{
    int num,ans;
    int son[3];
}t[maxn];


void ins(char *s,int k)
{
    int now=0;
    int len=strlen(s);
    for (int i=0;i<len;i++)
    {
        if (t[now].son[s[i]-'a']==0)
        {
            t[now].son[s[i]-'a']=++cnt1;
        }
        now=t[now].son[s[i]-'a'];
        t[now].ans=max(t[now].ans,k-t[now].num-1);
        t[now].num=k;
    }
}

int getans(char *s)
{
    int now=0;
    int len=strlen(s);
    for (int i=0;i<len;i++)
    {
        if (t[now].son[s[i]-'a']==0) return n;
        now=t[now].son[s[i]-'a'];
    }
    return max(t[now].ans,n-t[now].num);
}

int main()
{
    freopen("word.in","r",stdin);
    freopen("word.out","w",stdout);
    scanf("%d%d",&n,&m);
    k=0;
    for (int i=1;i<=n;i++)
    {
        scanf("%s",s);
        ins(s,i);
    }
    for (int i=1;i<=m;i++)
    {
        scanf("%s",s);
        printf("%d\n",getans(s));
    }
}