Hdu 5782 Cycle(拓展KMP+Hash)

来源:互联网 发布:小米电视怎么样 知乎 编辑:程序博客网 时间:2024/06/05 11:36
题目地址:http://acm.hdu.edu.cn/showproblem.php?pid=5782

思路:

1.若两串st1、st2循环同构(即两串首位相接旋转后相等),则必存在st1=str1+str2,st2=str2+str1(即串1的后缀等于串2的前缀,串1的前缀等于串2的后缀)。

2.枚举长度len,对于前缀s1、s2,使用扩展KMP求出串s1的后缀[i....len]与串s2的最大前缀长度l,若l+i>=len,则比较剩余部分是否相等(即s1的前缀是否等于s2的后缀,若相等,即为循环同构),比较是否相等可通过Hash实现。

#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>#define debugusing namespace std;typedef long long LL;typedef unsigned long long ULL;const int base=233;const int maxn=1e4+10;int len;int nt[maxn],ext[maxn];char st1[maxn],st2[maxn];ULL x[maxn],Hash[2][maxn];void getnext(char *T){    int i,length=strlen(T);    nt[0]=length;    for(i=0; i<length-1&&T[i]==T[i+1]; i++);    nt[1]=i;    int a=1;    for(int k=2; k<length; k++)    {        int p=a+nt[a]-1,l=nt[k-a];        if((k-1)+l>=p)        {            int j=(p-k+1)>0?(p-k+1):0;            while(k+j<length&&T[k+j]==T[j]) j++;            nt[k]=j,a=k;        }        else nt[k]=l;    }}void getext(char *S,char *T){    memset(nt,0,sizeof(nt));    getnext(T);    int Slen = strlen(S), Tlen = strlen(T), a = 0;    int MinLen = Slen>Tlen?Tlen:Slen;    while(a<MinLen && S[a]==T[a]) a++;    ext[0] = a, a = 0;    for(int k = 1; k < Slen; k++)    {        int p = a+ext[a]-1, L = nt[k-a];        if( (k-1)+L >= p )        {            int j = (p-k+1)>0? (p-k+1) : 0;            while(k+j<Slen && j<Tlen && S[k+j]==T[j] ) j++;            ext[k] = j;            a = k;        }        else ext[k] = L;    }}void init(){    x[0]=1;    for(int i=1; i<maxn; i++) x[i]=x[i-1]*base;}void preHash(int id,string s){    Hash[id][0]=0;    for(int i=0; i<len; i++) Hash[id][i]=Hash[id][i-1]*base+(s[i]-'a');}ULL getHash(int l,int r,int id){    return Hash[id][r]-Hash[id][l-1]*x[r-l+1];}int main(){#ifdef debu    freopen("in.txt","r",stdin);#endif // debug    init();    while(scanf("%s%s",st1,st2)!=EOF)    {        len=strlen(st1);        getext(st1,st2);        preHash(0,st1);        preHash(1,st2);        for(int i=1; i<=len; i++)        {            int flag=0;            for(int j=0; j<i; j++)            {                if(ext[j]+j>=i)                {                    if(getHash(0,j-1,0)==getHash(min(i-j,ext[j]),i-1,1))                    {                        flag=1;                        break;                    }                }            }            if(flag) printf("1");            else printf("0");        }        printf("\n");    }    return 0;}




原创粉丝点击