POJ 1509 环形后缀数组

来源:互联网 发布:linux jobs命令 编辑:程序博客网 时间:2024/06/06 01:08
////  main.cpp//  后缀数组////  Created by liuzhe on 17/2/5.//  Copyright © 2017年 my_code. All rights reserved.//#include <iostream>#include <cstdio>#include <cstring>#include <string>#include <algorithm>using namespace std;
//分析是学长的/*题意:给个串问最小的后缀,但是这个串是个环思路:是个环也好处理,那么就将串复制一遍加到后面就行了,
然后求出后缀数组和高度数组,刚开始我写的是就求出后缀数
组sa,然后从排名第1开始找,比如找到了I,排名为I的从第
sa[I]开始,那么如果这个值小于输入串的长度,直接输出就
行,提交WA,看了看讨论,有这样的数组,aaaa那么应该输出
1,而那个程序输出4,然后想到用高度数组,刚开始找到4的
位置,然后找下一个排名的高度数组,如果lcp的值大于串的长
度,那么更新结果,以此类推直到lcp的值不再大于原串的长度,
而为什么大于原串长度就可以继续呢,打个比方,abab,变成
abababab,sa数组为4 8 3 7 2 6 1 5,lcp数组为
0 2 4 6 0 1 3 5 ,然后我们找到的第一个pos为3,然后
它的下一个就是pos为1的位置,如果lcp[3]的值大于原串长度4,
就说明从1开始的串,可以代替从3开始的串,因为最终的长度肯
定为4嘛,在说的具体一点,看的时候拿纸画一下,从3开始的串
代表的是3 4 1 2 ,体现在复制一遍的串上就是3 4 5 6 ,而
从1开始的是1 2 3 4,所以如果lcp[3]大于4,代表3 4 5 6
与1 2 3 4是相同的串,懂了吗~~~~*/// poj 1509const int maxn = 50010;int wa[maxn],wb[maxn],wv[maxn],ww[maxn];int sa[maxn],lcp[maxn],rank[maxn],rank1[maxn];char str[maxn];int t,nn,n,k;int tmp[maxn];bool compare_sa(int i,int j){    if(rank1[i]!=rank1[j]) return rank1[i]<rank1[j];    else{        int ri=i+k<=n? rank1[i+k]:-1;        int rj=j+k<=n? rank1[j+k]:-1;        return ri<rj;    }}void construct_sa(){    for(int i=0;i<=n;i++){        sa[i]=i;        rank1[i]=i<n? str[i]:-1;    }    for(k=1;k<=n;k*=2){        sort(sa,sa+1+n,compare_sa);        tmp[sa[0]]=0;        for(int i=1;i<=n;i++){            tmp[sa[i]]=tmp[sa[i-1]]+(compare_sa(sa[i-1],sa[i])?1:0);        }        for(int i=0;i<=n;i++) rank1[i]=tmp[i];    }}/*inline bool cmp(int *r,int a,int b,int len){    return r[a]==r[b]&&r[a+len]==r[b+len];}void construct_sa(){    int m = 20010;    int i,j,p,*x=wa,*y=wb,*t;    //n++;//why?    for(i=0;i<m;i++) ww[i] = 0;    for(i=0;i<n;i++) ww[x[i]=str[i]]++;    for(i=1;i<m;i++) ww[i] += ww[i-1];    for(i=n-1;i>=0;i--) sa[--ww[x[i]]] = i;    for(j=p=1;p<n;j<<=1,m=p)    {        for(p=0,i=n-j;i<n;i++)            y[p++] = i;        for(i=0;i<n;i++)        {            if(sa[i]>=j)                y[p++] = sa[i]-j;        }        for(i=0;i<m;i++)            ww[i] = 0;        for(i=0;i<n;i++)            ww[wv[i]=x[y[i]]] ++;        for(i=1;i<m;i++)            ww[i]+=ww[i-1];        for(i=n-1;i>=0;i--)            sa[--ww[wv[i]]] = y[i];        for(t=x,x=y,y=t,x[sa[0]]=0,p=i=1;i<n;i++)            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;    }}*/void construct_lcp(){    for(int i=0;i<=n;i++)        rank1[sa[i]] = i;    int h = 0;    lcp[0] = 0;    for(int i=0;i<n;i++)    {        int j = sa[rank1[i]-1];        if(h>0)            h--;        for(;j+h<n && i+h<n;h++)            if(str[i+h]!=str[j+h])                break;        lcp[rank1[i]-1] = h;    }}int main(){    int T;    scanf("%d",&T);    while(T--){        scanf("%s",str);        int nn=strlen(str);        for(int i=0;i<nn;i++) str[nn+i]=str[i];        n=2*nn;        construct_sa();        construct_lcp();        int pos=n+1;        for(int i=1;i<=n;i++){            if(sa[i]<nn){                pos=sa[i++];                while(i<=n&&lcp[i-1]>=nn){                    pos=min(pos,sa[i]);i++;                }                break;            }        }        printf("%d\n",pos+1);    }    return 0;}/*int main(){    int T;    cin>>T;    while(T--)    {        scanf("%s",str);        int nn = strlen(str);        for(int i=0;i<nn;i++)            str[nn+i] = str[i];        int n = nn<<1;        construct_sa(n+1,200010);        construct_lcp(n);        int pos = n+1;        for(int i=1;i<=n;i++)        {            if(sa[i]<nn)            {                pos = sa[i++];                while(i<=nn&&lcp[i-1]>=nn)                {                    pos = min(pos,sa[i]);                    i++;                }                break;            }        }        printf("%d\n",pos+1);    }    return 0;}*/

0 0
原创粉丝点击