HDU 5782 Cycle（后缀数组+bitset）

Description
给定两个长度相等的字符串a和b，问他们的第 i个前缀是否循环相等，循环相等的定义是，两个长度相等的字符串，其中一个字符串能通过循环移位得到另一个
Input
多组用例，每组用例输入两个长度相等的字符串（串长不超过10000）
Output
对于每组用例，输出n个01值，第i个值表示两个串长度为i的前缀是否循环相等，相等输出1，不相等输出0
Sample Input
abc
cab
aa
aa
Sample Output
001
11
Solution
两串长度为i的前缀循环相等其实就是a[0,len)=b[i-len,i)，a[len,i)=b[0,i-len)
思路就是先存下b的任一前缀[0,i)在a中能够找到的匹配[j,j+i)，然后对于b的长度为i的前缀，求出a[0,n)与b[i,n)匹配的最大长度，判断这个长度是否大于等于j，如果大于等于j则说明满足循环相等的两个条件，否则不行
求a串和b串前后缀的最大匹配很简单，只要将a串b串用其他字符隔开然后连在一起，做一遍后缀数组后对height数组做一遍rmq即可O(1)查询得到
对于第一个问题，也就是求b的任一前缀[0,i)在a中能够找到的匹配[j,j+i)，如果枚举b的前缀显然不行，换个思路，对于a的每个后缀a[i,n)，让其与b串做匹配，如果长度为k，说明b串任一长度不超过k的前缀都可以与a[i,n)匹配，即若a[i,n)最多能和b[0,k)匹配，那么a[i,n)可以和b[0,1),b[0,2),…,b[0,k)匹配，具体实现就是游标法，第一个游标从大到小枚举匹配长度i，第二个游标移动a的后缀起始点j（先将匹配长度从大到小排序才能使用游标法），用temp（bitset）存储匹配长度大于等于i的j，每次j不能移动了就将这个temp值赋给vis[i]（vis[i]也是一个bitset，存满足b[0,i)=a[j,j+i)的j），注意这个temp值赋完后不要清空，因为这些j已经满足了一个较长的长度i，那么一定也满足较小的长度
对于第二个问题，枚举i，令len为a[0,n)和b[i,n)的最大匹配长度，通过vis[i]我们知道所有合法的j使得b[0,i)=a[j,j+i)，若j<=len，那么a[0,j)=b[i,j+i)，进而ans[i+j]=1，但是枚举j显然不行，还是得用bitset优化，预处理出一个数组pre，pre[i]是一个0~i位是1的bitset，假设答案存在ans中（还是个bitset），那么对于一个i，直接令ans|=(vis[i]<< i)&(pre[i+len]^pre[i-1])即找出了所有满足条件的j
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<bitset>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 22222
int t1[maxn],t2[maxn],sa[maxn],c[maxn],Rank[maxn],height[maxn];
bool cmp(int *r,int a,int b,int l)
{
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int str[],int n,int m)
{
    n++;
    int i,j,p,*x=t1,*y=t2;
    for(i=0;i<m;i++)c[i]=0;
    for(i=0;i<n;i++)c[x[i]=str[i]]++;
    for(i=1;i<m;i++)c[i]+=c[i-1];
    for(i=n-1;i>=0;i--)sa[--c[x[i]]]=i;
    for(j=1;j<=n;j<<=1)
    {
        p=0;
        for(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++)c[i]=0;
        for(i=0;i<n;i++)c[x[y[i]]]++;
        for(i=1;i<m;i++)c[i]+=c[i-1];
        for(i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1;x[sa[0]]=0;
        for(i=1;i<n;i++)
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
        if(p>=n)break;
        m=p;
    }
    int k=0;
    n--;
    for(i=0;i<=n;i++)Rank[sa[i]]=i;
        for(i=0;i<n;i++)
        {
            if(k)k--;
            j=sa[Rank[i]-1];
            while(str[i+k]==str[j+k])k++;
            height[Rank[i]]=k;
        }
}
int RMQ[maxn],mm[maxn],best[22][maxn];
void Init_rmq(int n)
{
    for(int i=1;i<=n;i++)RMQ[i]=height[i];
    mm[0]=-1;
    for(int i=1;i<=n;i++)
        mm[i]=((i&(i-1))==0)?mm[i-1]+1:mm[i-1];
    for(int i=1;i<=n;i++)best[0][i]=i;
    for(int i=1;i<=mm[n];i++)
        for(int j=1;j+(1<<i)-1<=n;j++)
        {
            int a=best[i-1][j];
            int b=best[i-1][j+(1<<(i-1))];
            if(RMQ[a]<RMQ[b])best[i][j]=a;
            else best[i][j]=b;
        }
}
int Ask_rmq(int a,int b)
{
    int t;
    t=mm[b-a+1];
    b-=(1<<t)-1;
    a=best[t][a];b=best[t][b];
    return RMQ[a]<RMQ[b]?a:b;
}
int lcp(int a,int b)
{
    a=Rank[a];b=Rank[b];
    if(a>b)swap(a,b);
    return height[Ask_rmq(a+1,b)];
}
bitset<maxn/2>pre[maxn/2],vis[maxn/2],temp,ans;
char x[maxn],y[maxn];
int a[maxn];
struct node
{
    int v,pos;
    bool operator <(const node &b)const
    {
        if(v!=b.v)return v>b.v;
        return pos<b.pos;
    }
}p[maxn];
int main()
{
    pre[0].reset();
    pre[0][0]=1;
    for(int i=1;i<maxn/2;i++)
        pre[i]=pre[i-1],pre[i].set(i);
    while(~scanf("%s%s",x,y))
    {
        int n=strlen(x);
        for(int i=0;i<n;i++)a[i]=x[i];
        a[n]='#';
        for(int i=n+1;i<=2*n;i++)a[i]=y[i-n-1];
        a[2*n+1]=0;
        da(a,2*n+1,222);
        Init_rmq(2*n+1);    
        for(int i=0;i<n;i++)
            p[i].v=lcp(i,n+1),p[i].pos=i;
        sort(p,p+n);
        temp.reset();
        for(int i=n,j=0;i>=1;i--)
        {
            while(j<n&&p[j].v>=i)temp.set(p[j].pos),j++;
            vis[i]=temp;
        }
        ans.reset();
        for(int i=1;i<=n;i++)
        {
            if(i==n)ans|=(vis[n]<<n);
            else
            {
                int len=lcp(0,n+i+1);
                ans|=((vis[i]<<i)&(pre[i+len]^pre[i-1]));
            }
        }
        for(int i=1;i<=n;i++)
            if(ans.test(i))printf("1");
            else printf("0");
        printf("\n");
    }
    return 0;
}