Cod sursa(job #3370662)

Utilizator NFJJuniorIancu Ivasciuc NFJJunior Data 26 octombrie 2026 17:53:32
Problema Potrivirea sirurilor Scor 14
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.27 kb
#include <bits/stdc++.h>
using namespace std;

vector<int> rabin_karp(string_view text, string_view pattern)
{
    int n = text.size();
    int m = pattern.size();

    if (n < m)
        return {};

    constexpr int P1 = 257;
    constexpr int MOD = 1e9 + 7;

    vector<int> p_pow1(n);
    p_pow1[0] = 1;
    for (int i = 1; i < n; i++) {
        p_pow1[i] = ((long long)p_pow1[i - 1] * P1) % MOD;
    }

    vector<int> h1(n + 1);
    h1[0] = 0;
    for (int i = 0; i < n; i++) {
        h1[i + 1] = ((long long)h1[i] + (long long)text[i] * p_pow1[i]) % MOD;
    }

    long long h_p1 = 0;
    for (int i = 0; i < m; i++) {
        h_p1 = (h_p1 + pattern[i] * p_pow1[i]) % MOD;
    }

    vector<int> indices;
    for (int i = 0; i + m - 1 < n; i++) {
        long long t1 = (h1[i + m] + MOD - h1[i]) % MOD;
        if (t1 == h_p1 * p_pow1[i] % MOD)
            indices.push_back(i);
    }
    return indices;
}

int main()
{
    freopen("strmatch.in", "r", stdin);
    freopen("strmatch.out", "w", stdout);

    string text;
    string pattern;
    cin >> pattern >> text;

    vector<int> indices = rabin_karp(text, pattern);
    cout << indices.size() << '\n';
    for (int i = 0; i < 1000 && i < (int)indices.size(); i++)
        cout << indices[i] << ' ';
    cout << '\n';

    return 0;
}