Cod sursa(job #1321521)

Utilizator badea_adi1999Badea Adrian Catalin badea_adi1999 Data 19 ianuarie 2015 11:28:10
Problema Potrivirea sirurilor Scor 78
Compilator cpp Status done
Runda Arhiva educationala Marime 1.44 kb
#include <cstdio>
#include <fstream>
#include <cstring>
#define p 73
#define nmax 2000005
#define mod1 666013
#define mod2 31115
using namespace std;
ifstream f("strmatch.in");
ofstream g("strmatch.out");
char a[nmax],b[nmax];
int h1,h2;
int ha1,ha2;
int n,m,c1,c2;
int match[2005],nr;

void Hashing()
{
    int i;
    c1=1;c2=1;
    for (i=0;i<n;i++)
                c1=(c1*p)%mod1;
    for (i=0;i<n;i++)
                c2=(c2*p)%mod2;
    for (i=0;i<n;i++)
                ha1=(ha1*p+(a[i]-'0')) %mod1;
    for (i=0;i<n;i++)
                ha2=(ha2*p+(a[i]-'0')) %mod2;

    h1=b[0]-'0';h2=b[0]-'0';

    //hashing dublu

    for (i=1;i<n;i++) {
                h1=(h1*p+ (b[i]-'0') )%mod1;
                h2=(h2*p+ (b[i]-'0') )%mod2;
    }
    for (i=n;i<=m;i++)
                {h1=(h1*p+ (b[i]-'0')-(b[i-n]-'0')*c1 )%mod1;
                 h2=(h2*p+ (b[i]-'0')-(b[i-n]-'0')*c2 )%mod2;
                 if (h1<0) h1+=mod1;
                 if (h2<0) h2+=mod2;
                 if (h1==ha1&&h2==ha2) {
                    nr++;
                    if (nr<=1000) match[nr]=i;
                 }
        }
}


int main()
{
    int i,j;
    f.getline(a,nmax);
    n=strlen(a);
    f.getline(b,nmax);
    m=strlen(b);

    Hashing();
    g<<nr<<'\n';
    if (nr<=1000)
            for (i=1;i<=nr;i++) g<<match[i]-n+1<<' ';
       else
            for (i=1;i<=1000;i++) g<<match[i]-n+1<<' ';


    return 0;
}