Cod sursa(job #2107723)

Utilizator Andrei243Nitu Mandel Andrei Andrei243 Data 17 ianuarie 2018 17:34:13
Problema Abc2 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.67 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[35000];


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')<<2*(marime_karp-i-1);
    }
    //cout<<sir<<' '<<rez<<'\n';
    return rez;
}

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

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=abs(key%25000);
    vector <unsigned long long>::iterator it;
    for(it=cuvinte[key].begin(); it!=cuvinte[key].end(); it++)
    {
        if(*it==aux)return 1;
    }

    return 0;
}


int main()
{
    in>>sir;


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


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

        if(cautare(cheie))nr++;

        cheie<<=2;
        cheie%=1<<(2*marime_karp);
        //cheie-=(unsigned long long)(sir[i]-'a')<<(2*marime_karp);
        cheie+=sir[i+marime_karp]-'a';

    }
    out<<nr;
    return 0;
}