Cod sursa(job #1539632)

Utilizator MarghescuGabriel Marghescu Marghescu Data 1 decembrie 2015 07:34:36
Problema Potrivirea sirurilor Scor 14
Compilator cpp Status done
Runda Arhiva educationala Marime 1.49 kb
#include <fstream>
#include <cstring>
#define M1 100007
#define M2 100021
using namespace std;
ifstream f("strmatch.in");
ofstream g("strmatch.out");
char a[2000001],b[2000001];
int v[1001];
int powr(int baza,int putere)
{
    int r=1;
    while(putere>1)
    {
           if(putere%2==0)
           {
               putere>>=1;
               baza=(baza*baza)%M1;
           }
           else
           {
               putere--;
               r=(r*baza)%M1;
           }
     }
    return (baza*r)%M1;
}
int main()
{
    int hashA=0,hashB=0,nr=0,hashA2=0,hashB2=0;
    f.getline(a,2000001);
    f.getline(b,2000001);
    int length=strlen(a);
    int power=length-1;
    int powermax=powr(256,power);
    for(int i=0;i<length;i++)
    {
        hashA=(hashA*256+(int)a[i])%M1;
        hashA2=(hashA2*256+(int)a[i])%M2;
    }
    for(int i=0;i<length;i++)
    {
        hashB=(hashB*256+(int)b[i])%M1;
        hashB2=(hashB2*256+(int)b[i])%M2;
    }
    int k=0;
    if(hashA==hashB&&hashA2==hashB2)
    {
        nr++;
        v[++k]=0;
    }
    int lengthB=strlen(b);
    for(int i=length;i<lengthB;i++)
    {
        hashB=((hashB-(b[i-length]*powermax)%M1+M1)*256+b[i])%M1;
        hashB2=((hashB2-(b[i-length]*powermax)%M2+M2)*256+b[i])%M2;
        if(hashB==hashA&&hashA2==hashB2)
        {
            nr++;
            v[++k]=i-length+1;
        }
    }
    g<<nr<<"\n";
    for(int i=1;i<=k&&k<=1000;i++)
        g<<v[i]<<" ";
    return 0;
}