Cod sursa(job #2172688)

Utilizator RaduMirceaAndreiRadu Mircea Andrei RaduMirceaAndrei Data 15 martie 2018 17:31:52
Problema Potrivirea sirurilor Scor 16
Compilator cpp Status done
Runda Arhiva educationala Marime 0.86 kb
# include <fstream>
# include <cstring>
# define DIM 2000010
using namespace std;
ifstream fin("strmatch.in");
ofstream fout("strmatch.out");
char a[DIM],b[DIM];
int sol[DIM],n,m,i,nr;
unsigned long long p[DIM],vala,valb;
int cod(char c){
    if(c>='A'&&c<='Z')
        return 26+c-'A';
    else
        return c-'a';
}
int main () {
    fin>>a+1>>b+1;
    n=strlen(a+1);
    m=strlen(b+1);
    p[0]=1;
    for(i=1;i<=n;i++)
        p[i]=p[i-1]*52;
    for(i=1;i<=n;i++)
        vala=vala*52+cod(a[i]);
    for(i=1;i<=n;i++)
        valb=valb*52+cod(b[i]);
    if(vala==valb)
        sol[++nr]=n-1;
    for(;i<=m;i++){
        valb-=p[n-1]*cod(b[i-n]);
        valb=valb*52+cod(b[i]);
        if(vala==valb)
            sol[++nr]=i-n;
    }
    fout<<nr<<"\n";
    for(i=1;i<=min(nr,1000);i++)
        fout<<sol[i]<<" ";
    return 0;
}