Z-array

input/code.cpp: In function 'int main()':
input/code.cpp:49:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   49 |     for (int i = 0; i < z_array.size(); i++) {
      |                     ~~^~~~~~~~~~~~~~~~
input/code.cpp:51:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   51 |         if (i < z_array.size() - 1) cout << " ";
      |             ~~^~~~~~~~~~~~~~~~~~~~

#include <iostream>
#include <vector>
#include <string>
 
using namespace std;
 
vector<int> calculateZ(string s) {
    int n = s.size();    // Tamaño del string
    vector<int> Z(n, 0); // Inicializar la Z-array con 0s
    Z[0] = n;            // Z[0] es siempre igual al tamaño del string
 
    int L = 0, R = 0;    // Inicializa los extremos de la ventana de coincidencia más reciente
    for (int i = 1; i < n; i++) {
        if (i > R) {
            // No hay coincidencia anterior que pueda reutilizarse
            L = R = i;
            // Comparación directa del substring desde i con el prefijo de s
            while (R < n && s[R] == s[R - L]) {
                R++;
            }
            Z[i] = R - L; // Longitud del substring que coincide con el prefijo
            R--;          // Ajustar R para que sea el índice del último carácter coincidente
        } else {
            int k = i - L; // Posición relativa en el substring que se está reutilizando
            // Si el valor de Z en la posición k no supera el final de la ventana actual
            if (Z[k] < R - i + 1) {
                Z[i] = Z[k]; // Podemos reutilizar el valor de Z[k]
            } else {
                // Necesidad de expandir la ventana
                L = i;
                while (R < n && s[R] == s[R - L]) {
                    R++;
                }
                Z[i] = R - L;
                R--;
            }
        }
    }
    return Z;
}
 
int main() {
    string s;
    cin >> s;                  // Leer el string de entrada
 
    vector<int> z_array = calculateZ(s);  // Calcular la Z-array
 
    // Imprimir la Z-array
    for (int i = 0; i < z_array.size(); i++) {
        cout << z_array[i];
        if (i < z_array.size() - 1) cout << " ";
    }
    cout << endl;
 
    return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED