HDU - 5782

給定兩個長度相等的字串
問他們的第 i個字首是否迴圈相等
迴圈相等的定義是，兩個長度相等的字串
其中一個字串能通過迴圈移位得到另一個

按照題解的說法，如果兩個字串迴圈相等
將 A串拆成字首和字尾，交換位置就能得到 B串

然後我對著題解懵逼了一小時，沒懂接下來怎麼做
還好 lcy打野強力輔助，我終於懂了一點

利用一個 bitset，先預處理出 26個字母在 A，B串的分佈
Ap[i][j]表示 A串長度為 i的字首能在 B串的 j位置找到匹配
Bp同理，Bp[i][j]為 B串長度為 i的字首能在 A串的 j位置匹配

接下在 A

有幾個注意點：

1. 二分要特判一下無解的情況
2. 要特判一下前 i位全部相等的情況

#pragma comment(linker, "/STACK:102400000,102400000")
#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
 

#include <algorithm>
#include <cmath>
#include <cctype>
#include <map>
#include <set>
#include <queue>
#include <bitset>
#include <string>
using namespace std;
typedef pair<int,int> Pii;
typedef long long LL;
typedef unsigned long long ULL;
typedef
 
 double DBL;
typedef long double LDBL;
#define MST(a,b) memset(a,b,sizeof(a))
#define CLR(a) MST(a,0)
#define SQR(a) ((a)*(a))
#define PCUT puts("\n----------")

const int maxn=1e4+10;
//const int maxn=10;

char A[maxn], B[maxn];
bitset<maxn> posA[26], posB[26], 
Ap[maxn], Bp[maxn], ans, tem, pre[maxn];

int main()
{
    #ifdef LOCAL
    freopen("in.txt", "r", stdin);
//    freopen("out.txt", "w", stdout);
    #endif

    for(int i=1; i<maxn; i++) pre[i] = pre[i-1].set(i-1);
    while(~scanf(" %s %s", A, B))
    {
        int N=strlen(A);
        for(int i=0; i<26; i++) posA[i].reset(), posB[i].reset();

        for(int i=0; i<N; i++)
        {
            posA[A[i]-'a'][i] = 1;
            posB[B[i]-'a'][i] = 1;
        }

        Ap[0] = posB[A[0]-'a'];
        Bp[0] = posA[B[0]-'a'];
        for(int i=1; i<N; i++) Bp[i] = Bp[i-1] & (posA[B[i]-'a']>>i);

        ans.reset(); tem.reset();
        bool all = A[0]==B[0];
        if(all) ans[0]=1;
        for(int i=1; i<N; i++)
        {
            Ap[i] = Ap[i-1] & (posB[A[i]-'a']>>i);
            int l=0, r=N-1;
            while(l<r)
            {
                int mid=(l+r+1)>>1;
                if(!Bp[mid][i]) r=mid-1;
                else l=mid;
            }
            all &= A[i]==B[i];
            if(all) ans[i]=1;
            if(l==0 && A[i] != B[0]) continue;
            ans |= (Ap[i-1]&pre[l+1]) << (i-1);
        }
        for(int i=0; i<N; i++) putchar(ans[i]+'0');
        puts("");
    }
    return 0;
}