[jzoj3126]【GDKOI2013选拔】大LCP(字典树解LCP)

来源:互联网 发布:淘宝评论在哪里看 编辑:程序博客网 时间:2024/05/18 03:39

这个题就不传送门了,估计大部分人都看不了。

Description

LCP就是传说中的最长公共前缀,至于为什么要加上一个大字,那是因为…你会知道的。

首先,求LCP就要有字符串。既然那么需要它们,那就给出n个字符串好了。

于是你需要回答询问大LCP,询问给出一个k,你需要求出前k个字符串中两两的LCP最大值是多少,这就是传说中的大LCP。

Input

第一行一个整数N,Q,分别表示字符串个数和询问次数。

接下来N行,每行一个字符串。

再Q行,每行一个正整数k。

Output

Q行,依次分别表示对Q个询问的答案。

Sample Input

3 3
a
b
ab
1
2
3

Sample Output

0
0
1

Data Constraint

对于30%的数据,字符串总长度不超过10^4,1<=N<=10^3,1<=Q<=10.

接下来30%的数据,字符串总长度不超过10^4,1<=N<=10^3,1<=Q<=1000.

对于100%的数据,字符串总长度不超过10^6,1<=N,Q<=10^5.


以下每看一句话请认真思考。
首先,LCP可以转化为在Trie树上求lca。
然后,我们就可以求两两之间的最大LCP了。
接着,我们就可以预处理出前1~n个字符串的最大LCP了。
最后,回答询问即可。
代码:

#include<cstdio>#include<cstring>#include<iostream>#include<cmath>#include<algorithm>#include<cstdlib>#define ll long longusing namespace std;inline int read(){    int x=0;char ch=' ';int f=1;    while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();    if(ch=='-')f=-1,ch=getchar();    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();    return x*f;}int q,tot;int ch[1000001][26];int fa[1000001];int val[1000001];int dep[1000001];int mx;inline void insert(char *s){    int n=strlen(s);int u=0;    for(int i=0;i<n;i++){        int c=s[i]-'a';        int &to=ch[u][c];        if(!to)to=++tot;        fa[to]=u;        dep[to]=dep[u]+1;        u=to;        val[u]++;    }    while(u){        if(val[u]>1){            mx=max(mx,dep[u]);            break;        }        u=fa[u];    }}int n;char s[1000001];int ans[100001];int main(){    n=read();q=read();    for(int i=1;i<=n;i++){        scanf("%s",s);        insert(s);        ans[i]=mx;    }    for(int i=1;i<=q;i++){        int k=read();        printf("%d\n",ans[k]);    }    return 0;}
原创粉丝点击