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

Description

Input

Output

Sample Input

5 5
zzj
pri
prime
ime
owaski
2 3 1 3 5
2 2 2 3
1 actri
2 2 3 4
2 3 2 6 5

Sample Output

0
0
3
1

Data Constraint

Hint

zzj，prime，owaski三种基因片段的最长共有有效片段为空
pri，prime两种基因片段的最长共有效片段为空
添加基因片段actri，编号为6
prime，ime两种基因片段的最长共有有效片段为ime
pri，actri，owaski三种基因片段的最长共有有效片段i

Solution

• 这题是 $Trie+Lca$ 的经典题。

• 对于加入的每个字符串，倒序放入一个 $Trie$ 中，记录其最终节点、深度、倍增ST表等信息。

• 那么对于查询的那几个串，只需求出它们在 $Trie$ 上的 $Lca$ 即可，答案即为 $Lca$ 深度。

• 时间复杂度 $O(M*T*log\ Len)$ 。

Code

#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=1e5+2,M=N*10;
int len,tot;
int s[M],w[N],p[N],g[M][26];
int f[M][20],dep[M],a[10];
inline int read()
{
    int X=0,w=1; char ch=0;
    while(ch<'0' || ch>'9') {if(ch=='-') w=-1;ch=getchar();}
    while(ch>='0' && ch<='9') X=(X<<3)+(X<<1)+ch-'0',ch=getchar();
    return X*w;
}
inline int write(int x)
{
    if(x>9) write(x/10);
    putchar(x%10+'0');
}
inline void insert(int x)
{
    char ch=getchar();
    while(ch<'a' || ch>'z') ch=getchar();
    while(ch>='a' && ch<='z') s[++len]=ch-'a',ch=getchar();
    w[x+1]=len+1;
}
inline void trie(int x)
{
    int num=0;
    for(int i=w[x+1]-1;i>=w[x];i--)
    {
        if(!g[num][s[i]])
        {
            g[num][s[i]]=++tot;
            dep[tot]=dep[num]+1;
            f[tot][0]=num;
            for(int j=1;j<20;j++) f[tot][j]=f[f[tot][j-1]][j-1];
        }
        num=g[num][s[i]];
    }
    p[x]=num;
}
inline int lca(int x,int y)
{
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=log2(dep[x]);i>=0;i--)
        if(dep[f[x][i]]>=dep[y]) x=f[x][i];
    if(x==y) return x;
    for(int i=log2(dep[x]);i>=0;i--)
        if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
int main()
{
    int n=read(),m=read();
    for(int i=w[1]=1;i<=n;i++)
    {
        insert(i);
        trie(i);
    }
    while(m--)
    {
        int op=read();
        if(op==1)
        {
            insert(++n);
            trie(n);
        }else
        {
            int t=read();
            for(int i=0;i<t;i++) a[i]=read();
            sort(a,a+t);
            t=unique(a,a+t)-a;
            int ans=p[a[0]];
            for(int i=1;i<t;i++) ans=lca(ans,p[a[i]]);
            write(dep[ans]),putchar('\n');
        }
    }
    return 0;
}