Z-array

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

#include <iostream>
#include <vector>
#include <string>
 
using namespace std;
 
vector<long long> calculateZ(const string &s) {
    long long n = s.length();
    vector<long long> Z(n);
    long long l = 0, r = 0; 
 
    for (long long i = 1; i < n; i++) {
        if (i > r) {
            l = r = i;
            while (r < n && s[r] == s[r - l]) {
                r++;
            }
            Z[i] = r - l;
            r--;
        } else {
            long long k = i - l; 
            if (Z[k] < r - i + 1) {
                Z[i] = Z[k];
            } else {
                l = i;
                while (r < n && s[r] == s[r - l]) {
                    r++;
                }
                Z[i] = r - l;
                r--;
            }
        }
    }
    Z[0] = n;
    return Z;
}
 
int main() {
    string s;
    cin >> s;
 
    vector<long long> Z = calculateZ(s);
 
    for (long long i = 0; i < Z.size(); i++) {
        cout << Z[i];
        if (i < Z.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