Cod sursa(job #3214361)

Utilizator PalcPalcu Stefan Palc Data 14 martie 2024 08:45:25
Problema Potrivirea sirurilor Scor 38
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.21 kb
#include <bits/stdc++.h>
#define P 123457
#define Q 777013
using namespace std;

ifstream fin("strmatch.in");
ofstream fout("strmatch.out");

int n, m, cnt, codA1, codA2, codB1, codB2, p1, p2;
vector<int> sol;
char a[2000005], b[2000005];

int main()
{
    int i;
    fin >> a;
    fin >> b;
    n = strlen(a);
    m = strlen(b);
    p1 = p2 = 1;
    for (i = 0; i < n; i++)
    {
        codA1 = (codA1 * 62 + a[i]) % P;
        codA2 = (codA2 * 62 + a[i]) % Q;
        codB1 = (codB1 * 62 + b[i]) % P;
        codB2 = (codB2 * 62 + b[i]) % Q;
        if (i > 0)
        {
            p1 = (p1 * 62) % P;
            p2 = (p2 * 62) % Q;
        }
    }
    if (codA1 == codB1 && codA2 == codB2)
    {
        cnt++;
        sol.push_back(1);
    }
    for (i = n; i < m; i++)
    {
        codB1 = (codB1 - b[i - n] * p1 % P + P) % P;
        codB2 = (codB2 - b[i - n] * p2 % Q + Q) % Q;
        codB1 = (codB1 * 62 + b[i]) % P;
        codB2 = (codB2 * 62 + b[i]) % Q;
        if (codA1 == codB1 && codA2 == codB2)
        {
            cnt++;
            sol.push_back(i - n + 1);
        }
    }
    fout << cnt << "\n";
    for (int e : sol)
        fout << e << " ";
    fout << "\n";
    fin.close();
    fout.close();
    return 0;
}