最长公共前缀(lca+trie树)

问题描述

给出一些串,多组询问求两个串的最长公共前缀。字符串总长 < 10^6。

输入格式

第一行一个整数n,表示字符串的个数。

接下来n行,每行一个字符串(字符串不含空格)。

第n+2行一个整数m,表示询问总数。

接下来m行,每行两个整数a,b,表示询问第a个字符串和第b个字符串的最长公共前缀的长度。

输出格式

输出共m行,对于每个询问输出最长的公共前缀的长度

输入样例

5
abcdef
abcd
acd
cade
abcdef
4
1 2
2 3
1 5
3 4

输出样例

4
1
6
0

限制与约定

1<= m < 10^6

时间限制:1s

空间限制:256MB

#include
#include
#include
#include
#include
using namespace std;
const int maxn = 1000000 + 10 ;
int n,ch[maxn][30],val[maxn],cnt,m;
int f[maxn][31],Dep[maxn];
string str[30010];
void dfs(int u,int fa)
{
    Dep[u]=Dep[fa]+1;
    for(int i=0;i<=29;i++) f[u][i+1]=f[f[u][i]][i];
    for(int i=0;i<=26;i++)
    {
        int v = ch[u][i];
        if(!v) continue;
        f[v][0]=u;
        dfs(v,u);
    }
}
void insert(int pos)
{
    int len = str[pos].length(),u=0;
    for(int i=0;i)
    {
        int x = str[pos][i]-'a';
        if(!ch[u][x]) ch[u][x]=++cnt;
        u=ch[u][x];
    }
    val[pos]=u;
}
int LCA(int x,int y)
{
    if(Dep[x]<Dep[y]) swap(x,y);
    for(int i=30;i>=0;i--)
    {
        if(Dep[f[x][i]]>=Dep[y]) x=f[x][i];
        if(x==y) return x;
    }
    for(int i=30;i>=0;i--)
        if(f[x][i]!=f[y][i])
        {
            x=f[x][i];
            y=f[y][i];
        }
    return f[x][0];
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>str[i];
        insert(i);
    }
    dfs(0,-1);
    cin>>m;
    for(int i=1,a,b;i<=m;i++)
    {
        scanf("%d%d",&a,&b);
        int lca=LCA(val[a],val[b]);
        printf("%d\n",Dep[lca]-1);
    }
    return 0;
}

 

转载于:https://www.cnblogs.com/hfang/p/11047418.html

你可能感兴趣的:(最长公共前缀(lca+trie树))