JZOJ5397. 【NOIP2017提高A组模拟10.6】Biology

题解

如果我们将全部字符串倒过来，
那么后缀就变成了前缀。

如果用这些倒过来的字符串的建一棵tire，
最长公共前缀就是每一个字符串的最后一个点的lca的深度。

查询就用倍增+lca实现，
对应每加入一个新的节点，就处理一下它的父亲。

code

#include<queue>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#include <math.h>
#define ll long long
#define N 1000003
#define db double
#define P putchar
#define G getchar
#define mo 998244353
using namespace std;
char ch;
void read(int &n)
{
    n=0;
    ch=G();
    while((ch<'0' || ch>'9') && ch!='-')ch=G();
    ll w=1;
    if(ch=='-')w=-1,ch=G();
    while('0'<=ch && ch<='9')n=(n<<3)+(n<<1)+ch-'0',ch=G();
    n*=w;
}

ll max(ll a,ll b){return a>b?a:b;}
ll min(ll a,ll b){return a<b?a:b;}
ll abs(ll x){return x<0?-x:x;}
void write(ll x){if(x>9) write(x/10);P(x%10+'0');}

struct node
{
    int son[26];
}tr[N];

int n,m,f[N][20],deep[N];
int tot,t,k,a[N],x,pos[N];
char s[N];

int lca(int x,int y)
{
    if(deep[x]>deep[y])swap(x,y);
    for(int j=19;j>=0;j--)
        if(deep[x]<=deep[f[y][j]])y=f[y][j];
    if(x==y)return x;
    for(int j=19;j>=0;j--)
        if(f[x][j]!=f[y][j])x=f[x][j],y=f[y][j];
    return f[x][0];
}

int ins(char* s,int len)
{
    int x=1;
    for(int p=1;p<=len;p++)
    {
        if(tr[x].son[s[p]-'a']==0)
        {
            deep[++tot]=deep[x]+1;
            tr[x].son[s[p]-'a']=tot;
            f[tot][0]=x;
            for(int j=1;j<20;j++)
                f[tot][j]=f[f[tot][j-1]][j-1];
        }
        x=tr[x].son[s[p]-'a'];
    }
    return x;
}

int main()
{
    freopen("biology.in","r",stdin);
    freopen("biology.out","w",stdout);
    read(n);read(m);deep[1]=tot=1;
    for(int i=1;i<=n;i++)
    {
        ch=G();
        while(ch<'a' || ch>'z')ch=G();x=0;
        while('a'<=ch && ch<='z')s[++x]=ch,ch=G();
        for(int j=1;j<=x/2;j++)
            swap(s[j],s[x-j+1]);
        pos[i]=ins(s,x);
    }
    for(int i=1;i<=m;i++)
    {
        read(t);
        if(t==1)
        {
            ch=G();
            while(ch<'a' || ch>'z')ch=G();x=0;
            while('a'<=ch && ch<='z')s[++x]=ch,ch=G();
            for(int j=1;j<=x/2;j++)
                swap(s[j],s[x-j+1]);
            pos[++n]=ins(s,x);
        }
        else
        {
            read(k);read(t);t=pos[t];
            for(int j=2;j<=k;j++)
                read(x),t=lca(t,pos[x]);
            write(deep[t]-1),P('\n');
        }
    }
}