Cod sursa(job #3155676)

Utilizator BuzdiBuzdugan Rares Andrei Buzdi Data 9 octombrie 2023 11:34:07
Problema Prefix Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.16 kb
#include <fstream>
#include <vector>

using namespace std;

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

void test_case()
{
    string s;
    cin >> s;

    int n = (int) s.size();
    vector<int> LPS(n);

    LPS[0] = 0;
    for(int i = 1; i < n; i++)
    {
        int j = LPS[i - 1];
        while(j > 0 && s[i] != s[j])
            j = LPS[j - 1];
        if(s[i] == s[j])
            j++;
        LPS[i] = j;
    }

    bool terminat = false;
    int i = n - 1;
    while(!terminat && i > 0)
    {
        if(LPS[i] != 0)
        {
            int pos = i;
            int diff = i - LPS[i] + 1;
            int steps = 0;
            while(pos - LPS[pos] + 1 == diff && LPS[pos] != 0)
            {
                pos = pos - diff;
                steps++;
            }
            if(pos + 1 == diff)
            {
                steps++;
                cout << steps * diff << '\n';
                terminat = true;
            }
        }
        i--;
    }
    if(!terminat)
       cout << 0 << '\n';
}

int main()
{
    int T;
    cin >> T;
    while(T--)
        test_case();


    return 0;
}