Cod sursa(job #2131883)

Utilizator Constantin.Dragancea Constantin Constantin. Data 15 februarie 2018 08:42:41
Problema Potrivirea sirurilor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.41 kb
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

#define MOD 100007
#define mod2 100027
#define prime 73
//puterile descresc de la stanga la dreapta
//folosim 2 hashuri pentru a evita coliziunile

string a, b;
ll n, m, hasha, hasha2, hashb2, hashb, k, put = 1, put2 = 1;
vector <int> sol;

int main(){
    ifstream cin ("strmatch.in");
    ofstream cout ("strmatch.out");
    cin >> a >> b;
    n = a.length();
    m = b.length();
    for (int i=0; i<n; i++){
        //hashuim primu string in 2 hashuri
        hasha = (hasha * prime + a[i])% MOD;
        hasha2 = (hasha2 * prime + a[i])% mod2;

        //hashuim primele n caractere a sirului b in 2 hashuri
        hashb = (hashb * prime + b[i])% MOD;
        hashb2 = (hashb2 * prime + b[i])% mod2;

        //calculam puterea max (m-1)
        if (i){
            put = (put * prime)% MOD;
            put2 = (put2 * prime)% mod2;
        }
    }
    for (int i=0; i<=m-n; i++){
        if (hasha == hashb && hasha2 == hashb2){
            k++;
            if (k <= 1000) sol.push_back(i);
        }

        //facem update hashului subsirului din b, x = x-char1*put + newcar
        hashb = ((hashb - (b[i] * put) % MOD + MOD) * prime + b[i+n])% MOD;
        hashb2 = ((hashb2 - (b[i] * put2)%mod2 + mod2) * prime + b[i+n])% mod2;
    }
    cout << k << "\n";
    for (auto it: sol) cout << it << " ";
    return 0;
}