5pts求调

#include <bits/stdc++.h>
using namespace std;
string s;
string t;
int m;
int n;
int nxt[1000006];
void make()
{
    nxt[1]=0;
    int j = nxt[1];
    for(int i = 2;i<=m;i++)
    {
        while(t[j+1]!=t[i]&&j!=0) j=nxt[j];
        if(t[j+1]==t[i]) j++;
        nxt[i]=j;
    }
}
int match()
{
    int j = 0,ret = 0;
    for(int i = 1;i<=n;i++)
    {
        while(j!=0&&s[i]!=t[j+1])
        {
            j = nxt[j];
        }
        if(s[i]==t[j+1])
        {
            j++;
        }
        if(j==m)
        {
            return i-m+1;
        }
    }
}
int main()
{
    int ans = 0;
    cin>>n>>s;
    t = s;
    m = n;
    make();
    for(int i = 1;i<n;i++)
    {
        for(int j = 1;j<i;j++)
        {
            m=i-j;
            int k = match();
            if(k==1)
            {
                ans+=i-j;
                break;
            }
        }
    }
    cout<<ans<<"\n";
    return 0;
}