[Codeforces Gym] 100162B Circle of Stones

题意：

    桌子上有 n 个石头围成一个环。每个石头都有一种颜色。每种颜色可以由小写英文字母表示。如果每一对相邻的石头都是不同颜色的，则称这 n 个石头构成的环是美丽的。现在，你可以从这 n 个石头中拿走一段连续的石头（可以为空），且你只能拿一次。

　对于每个 k(0≤k≤n−1) ，判断是否存在一种取石头的方案，使得在拿走 k 个连续的石头后，剩下的 n−k 个石头构成的环是美丽的。

　n<=1000000

　正解kmp，转换思路，不去考虑拿走什么，考虑留下什么，不去考虑什么合法，去考虑哪些是违法的。

　显然如果a[i]=a[i+1]，a[i]与a[i+1]一定不能同时留下，所以我们可以将序列切成很多段，在每一段中，如果留下长度k不合法，那么说明k-1是它的一个循环节，因为不合法只能是头尾相同，即a[i]=a[i+k-1]，如果对于每个位置都满足，那么说明k-1是串的一个循环节(不要求乘除)。

　每一段求完再求个并就好了。

　next数组一定要从1开始！！！

　

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1000005
using namespace std;
char s[N*2];
int nxt[2*N],ans[2*N];
int n;
void solve(int l,int r)
{
    nxt[1]=0;int k=0;
    for(int i=l+1;i<=r;i++)
    {
        while(k&&s[k+l]!=s[i])k=nxt[k];
        if(s[k+l]==s[i])k++;
        nxt[i-l+1]=k;
    }
    int len=(r-l+1);
    for(int i=1;i<=len;i++)ans[i]++;
    while(k)
    {
        ans[len-k+1]--;
        k=nxt[k];
    }
    return ;
}
int main()
{
    int tot=0;
    while(~scanf("%s",s))
    {
        n=strlen(s);
        memset(ans,0,sizeof(ans));
        memset(nxt,0,sizeof(nxt));
        for(int i=0;i<n;i++)s[i+n]=s[i];
        int pre=0;
        for(int i=1;i<2*n;i++)
        {
            if(s[i]==s[i-1])
            {
                solve(pre,i-1);
                pre=i;
            }
        }
        solve(pre,2*n-1);
        printf("Case %d: ",++tot);
        for(int i=0;i<=n-1;i++)
        {
            if(ans[n-i])putchar('1');
            else putchar('0');
        }
        puts("");
    }
    return 0;
}