Cod sursa(job #2985210)

Utilizator pifaDumitru Andrei Denis pifa Data 25 februarie 2023 21:44:44
Problema Prefix Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.87 kb
#include <bits/stdc++.h>

using namespace std;

ifstream in("prefix.in");
ofstream out("prefix.out");

const int N = 1e6 + 5;

int pi[N];
signed main()
{
    int t;
    in >> t;
    while(t--)
    {
        string s;
        in >> s;
        int n = s.size();
        s = s + '#';
        pi[1] = 0;
        for (int i = 1; i < n; i++)
        {
            int j = pi[i-1];
            while (j > 0 && s[i] != s[j])
                j = pi[j-1];
            if (s[i] == s[j])
                j++;
            pi[i] = j;
        }
        int ans = 0;
        for(int i = 1; i <= n - 1; i++)
        {
            if(pi[i] > 0 && (i + 1) % (i + 1 - pi[i]) == 0)
                ans = i;
        }
        if(ans == 0)
            out << 0 << '\n';

        else out << ans + 1 << '\n';
        memset(pi, 0, sizeof(pi));
    }
    return 0;
}