CF961F k-substrings

CF961F k-substrings

题意：给出长度为nn的字符串T，对于每一个k子串，找出最大长度的t，使得t是k子串的前后缀，且t长度为奇数

思路：

从T串的中心开始向两边扩展。首先处理T串中心的k子串：如果T串长度为奇数，ans=-1;如果长度为偶数，ans=1或-1。

接着开始向两边扩展。下一个的最长前后缀长度最多是上一个的长度+2，最小是-1 。枚举每一种情况，用hash来判断是否能成立。

#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
const int N=1e6+7;
const int P=107;
const ll mod=1e9+7;
int n;
char s[N];
int ans[N];
int cnt,l,r;
ll x,y;
ll p[N],h[N];

void Hash(){
    p[0]=1;
    for(int i=1;i<=n;i++){
        p[i]=p[i-1]*P%mod;
        h[i]=(h[i-1]*P%mod+s[i]-'a'+1)%mod;
    }
}

ll hs(int l,int r){
    return ((h[r]-h[l-1]*p[r-l+1]%mod+mod)+mod)%mod;
}

int main(){
    scanf("%d%s",&n,s+1);
    Hash();
    cnt=n+1>>1;
    if(n&1)ans[cnt]=-1,l=r=cnt;
    else{
        l=n>>1,r=l+1;
        if(s[l]==s[r])ans[cnt]=1;
        else ans[cnt]=-1;
    }
    
    for(int i=cnt-1;i>=1;i--){
        l--,r++;
        for(int j=ans[i+1]+2;j>=-1;j-=2){
            x=hs(l,l+j-1),y=hs(r-j+1,r);
            if(x==y || j==-1){
                ans[i]=j;
                break;
            }
        }
    }
    for(int i=1;i<=cnt;i++)
        printf("%d ",ans[i]);
    return 0;
}