Cod sursa(job #3226821)

Utilizator _andrei4567Stan Andrei _andrei4567 Data 22 aprilie 2024 21:01:34
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.85 kb
#include <fstream>
#include <vector>

using namespace std;

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

const int N = 5e6;
int z[N + 1];

vector <int> sol;

string a, s;

int nr;

int main()
{
    cin >> a >> s;
    s = a + '@' + s;
    for (int i = 1, l = 0, r = 0; i < s.size(); ++i)
    {
        if (i <= r)
            z[i] = min (z[i - l], r - i + 1);
        while (i + z[i] < s.size() && s[i + z[i]] == s[z[i]])
            ++z[i];
        if (i + z[i] - 1 > r)
            r = i + z[i] - 1, l = i;
    }
    for (int i = a.size() + 1; i < s.size(); ++i)
    {
        if (z[i] == a.size())
        {
            ++nr;
            if (nr <= 1000)
                sol.push_back(i - a.size() - 1);
        }
    }
    cout << nr << '\n';
    for (auto it : sol)
        cout << it << ' ';
    return 0;
}