Codeforces 526D Om Nom and Necklace （KMP）

#include<iostream>
#include<cmath>
#include<cstring>
#include<queue>
#include<vector>
#include<cstdio>
#include<algorithm>
#include<map>
#include<set>
#define rep(i,e) for(int i=0;i<(e);i++)
#define rep1(i,e) for(int i=1;i<=(e);i++)
#define repx(i,x,e) for(int i=(x);i<=(e);i++)
#define X first
#define Y second
#define PB push_back
#define MP make_pair
#define mset(var,val) memset(var,val,sizeof(var))
#define scd(a) scanf("%d",&a)
#define scdd(a,b) scanf("%d%d",&a,&b)
#define scddd(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define pd(a) printf("%d\n",a)
#define scl(a) scanf("%lld",&a)
#define scll(a,b) scanf("%lld%lld",&a,&b)
#define sclll(a,b,c) scanf("%lld%lld%lld",&a,&b,&c)
#define IOS ios::sync_with_stdio(false);cin.tie(0)

using namespace std;
typedef long long ll;
template <class T>
void test(T a){cout<<a<<endl;}
template <class T,class T2>
void test(T a,T2 b){cout<<a<<" "<<b<<endl;}
template <class T,class T2,class T3>
void test(T a,T2 b,T3 c){cout<<a<<" "<<b<<" "<<c<<endl;}
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3fll;
const ll mod = 200907;
int T;
void testcase(){
    printf("Case #%d: ",++T);
}
const int MAXN = 1e6+5;
const int MAXM = 30;
int n,k;
char s[MAXN];
int Next[MAXN];

void getNext(){
    int i,j;
    j=Next[0]=-1;
    i=0;
    while(i<n){
        while(-1!=j&&s[i]!=s[j]) j=Next[j];
        Next[++i]=++j;
    }
}
int check(int i){
    int d=i-Next[i]; //最小循环节
    int cnt = i/d; //循环节次数
    if(i%d==0){ //被整除
        return cnt/k-cnt%k>=0;  //
    }else return cnt/k-cnt%k>0;

}
int main() {
#ifdef LOCAL
    freopen("in.txt","r",stdin);
#endif // LOCAL
    scdd(n,k);
    scanf("%s",s);
    mset(Next,0);
    getNext();
    for(int i=1;i<=n;i++){
        printf("%d",check(i));
    }
//    for(int i=0;i<=1+n;i++) printf("%d ",Next[i]);
    return 0;
}