Cod sursa(job #2097242)

Utilizator mihailrazMihail Turcan mihailraz Data 30 decembrie 2017 19:33:48
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.43 kb
#include <bits/stdc++.h>
#define LMAX 2000000
#define BAZA 75
#define MOD1 100007
#define MOD2 100021

using namespace std;
ifstream fi("strmatch.in");
ofstream fo("strmatch.out");
int val1,val2,rhash1,rhash2,lgt,lgp,putere1=1,putere2=1;
char A[LMAX+1],B[LMAX+1];
vector <int> V;

void Rabin_Karp(char T[], char P[])
{
    lgt=strlen(T);
    lgp=strlen(P);
    for(int i=1; i<=lgp-1; i++)
    {
        putere1*=BAZA;
        putere2*=BAZA;
        putere1%=MOD1;
        putere2%=MOD2;
    }
    for(int i=0; i<lgp; i++)
    {
        val1=(val1*BAZA%MOD1+(P[i]-'0'))%MOD1;
        val2=(val2*BAZA%MOD2+(P[i]-'0'))%MOD2;
    }
    for(int i=0; i<lgp; i++)
    {
        rhash1=(rhash1*BAZA%MOD1/*%MOD2*/+(T[i]-'0'))%MOD1/*%MOD2*/;
        rhash2=(rhash2*BAZA%MOD2/*%MOD2*/+(T[i]-'0'))%MOD2/*%MOD2*/;
    }
    if(rhash1==val1 && rhash2==val2)
        V.push_back(0);
    for(int i=lgp; i<lgt; i++)
    {
        rhash1=(((rhash1-putere1*(T[i-lgp]-'0'))%MOD1+MOD1)*BAZA+(T[i]-'0'))%MOD1;
        rhash2=(((rhash2-putere2*(T[i-lgp]-'0'))%MOD2+MOD2)*BAZA+(T[i]-'0'))%MOD2;
        if(rhash1==val1 && rhash2==val2)
            V.push_back(i-lgp+1);
    }
}

void afisare(void)
{
    int nr=V.size();
    fo<<nr<<"\n";
    for(int i=0; i<min(nr,1000); i++)
        fo<<V[i]<<" ";
}

int main()
{
    fi>>A;
    fi>>B;
    Rabin_Karp(B,A);
    afisare();
    fi.close();
    fo.close();
    return 0;
}