Codeforces Round #542 [Alex Lopashev Thanks-Round] (Div. 1) C(二分＋KMP)

http://codeforces.com/contest/1129/problem/C

#include<bits/stdc++.h>
#define fi first
#define se second
#define INF 0x3f3f3f3f
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define pqueue priority_queue
#define NEW(a,b) memset(a,b,sizeof(a))
#define lowbit(x) (x&(-x))
#define si(x) scanf("%d",&x)
#define sl(x) scanf("%lld",&x)
#define lc (d<<1)
#define rc (d<<1|1)
const double pi=4.0*atan(1.0);
const double e=exp(1.0);
const int maxn=1e6+;
typedef long long LL;
typedef unsigned long long ULL;
const LL mod=1e9+;
const ULL base=1e7+;
using namespace std;
LL a[maxn],b[maxn],Next[maxn];
LL dp[][];
void get_next(int m){
    Next[]=-;
    int k=-;
    int j=;
    while(j<m){
        while(k>-&&b[k]!=b[j]){
            k=Next[k];
        }
        if(b[k]==b[j]||k==-){
            k++;
        }
        Next[++j]=k;
    }
}
bool kmp(int n,int m){
    int k=;
    for(int i=n-m+;i<=n;i++){
        b[k++]=a[i];
    }
    get_next(m);
    k=;
    for(int i=;i<n;i++)
    {
        while(k>&&b[k]!=a[i]){
            k=Next[k];
        }
        if(b[k]==a[i]){
            k++;
        }
        if(k==m){
            return ;
        }
    }
    return ;
}
bool check(int l){
    if(a[l]==&&a[l+]==&&a[l+]==&&a[l+]==) return ;
    if(a[l]==&&a[l+]==&&a[l+]==&&a[l+]==) return ;
    if(a[l]==&&a[l+]==&&a[l+]==&&a[l+]==) return ;
    if(a[l]==&&a[l+]==&&a[l+]==&&a[l+]==) return ;
    return ;
}
int main(){
    int n;
    LL sum=;
    scanf("%d",&n);
    for(int i=;i<n;i++){
        scanf("%d",&a[i]);
        if(i==){
            dp[][]=;
            printf("1\n");
        }
        else{
            int l=,r=i;
            int ans=;
            while(l<=r){
                int mid=(l+r)>>;
                if(kmp(i,mid)){
                    l=mid+;
                    ans=max(ans,mid);
                }
                else{
                    r=mid-;
                }
            }
            dp[i][i]=;
            if(i>) dp[i-][i]=;
            if(i>) dp[i-][i]=;
            if(i>&&check(i-)) dp[i-][i]=;
            for(int j=i-;j>=;j--){
                dp[j][i]+=dp[j][i-]*dp[i-][i];
                dp[j][i]%=mod;
            }
            for(int j=i-;j>=;j--){
                dp[j][i]+=dp[j][i-]*dp[i-][i];
                dp[j][i]%=mod;
            }
            for(int j=i-;j>=;j--){
                dp[j][i]+=dp[j][i-]*dp[i-][i];
                dp[j][i]%=mod;
            }
            for(int j=i-;j>=;j--){
                dp[j][i]+=dp[j][i-]*dp[i][i];
                dp[j][i]%=mod;
            }
            for(int j=i-ans;j>=;j--){
              sum+=dp[j][i];
              sum%=mod;
            }
            printf("%lld\n",sum);
        }
    }
}