Cod sursa(job #2107916)

Utilizator Andrei243Nitu Mandel Andrei Andrei243 Data 17 ianuarie 2018 19:54:32
Problema Abc2 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.78 kb
#include <iostream>
#include <fstream>
#include <cmath>
#include <cstring>
#include <vector>
using namespace std;

ifstream in("abc2.in");
ofstream out("abc2.out");

int marime_karp;
char sir[10001000];
vector<unsigned long long> cuvinte[500000];
unsigned long long puteri[25];

unsigned long long rabin_karp(char* sir)
{
    long long rez=0;
    for(int i=marime_karp-1; i>=0; i--)
    {
        rez+=(int)(sir[i]-'a')*puteri[marime_karp-i-1];
    }
    //cout<<sir<<' '<<rez<<'\n';
    return rez;
}

int hash1(char* sir)
{
    return rabin_karp(sir)%500000;
}

void adaugare(char* s)
{
    unsigned long long aux=rabin_karp(s);
    int aux_con=hash1(s);
    vector <unsigned long long>::iterator it;

    for(it=cuvinte[aux_con].begin(); it!=cuvinte[aux_con].end(); it++)
    {
        if(*it==aux)return;
    }
    cuvinte[aux_con].push_back(rabin_karp(s));
}

int cautare(unsigned long long key)
{
    unsigned long long aux=key;
    key=key%500000;
    vector <unsigned long long>::iterator it;
    for(it=cuvinte[key].begin(); it!=cuvinte[key].end(); it++)
    {
        if(*it==aux)return 1;
    }

    return 0;
}


void generare_puteri(){
puteri[0]=1;
for(int i=1;i<25;i++)puteri[i]=puteri[i-1]*4;
}

int main()
{
    generare_puteri();
    in>>sir;

    while(1)
    {
        char s[25];
        in>>s;
        marime_karp=strlen(s);
        adaugare(s);
        if(in.eof())break;


    }
    long nr=0;
    int aux=strlen(sir);
    unsigned long long cheie=rabin_karp(sir);
    for(int i=0; i<=aux-marime_karp; i++)
    {

        if(cautare(cheie))nr++;
        cheie<<=2;
        cheie^=(unsigned long long)(sir[i]-'a')*puteri[marime_karp];
        cheie^=sir[i+marime_karp]-'a';


    }
    out<<nr;
    return 0;
}