Cod sursa(job #1727346)

Utilizator cristina_borzaCristina Borza cristina_borza Data 10 iulie 2016 16:39:27
Problema Abc2 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.37 kb
#include <algorithm>
#include <fstream>
#include <cstring>
#include <set>
#include <map>

#define MOD 1000000007
#define BASE 3

using namespace std;

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

char cuv[50] , prop[10000005];
unsigned int n , m , nr;

int v[50005];

bool caut_bin(int val) {
    int i , p = 0;
    for (i = 1; i <= nr; i <<= 1);
    while (i) {
        if (i + p <= nr && v[i + p] <= val) {
            p += i;
        }

        i >>= 1;
    }

    if (v[p] == val) {
        return 1;
    }

    return 0;
}

int main() {
    f >> prop;
    m = strlen(prop);

    while (f >> cuv) {
        unsigned int hash1 = 0;
        n = strlen(cuv);

        for (int i = 0; i < n; ++i) {
            hash1 = (hash1 * BASE + (cuv[i] - 'a')) % MOD;
        }

        v[++nr] = hash1;
    }

    sort(v + 1, v + nr + 1);

    unsigned int P1 = 1, hash1 = 0 , sol = 0;
    for (int i = 0; i < n; ++i) {
        hash1 = (hash1 * BASE + (prop[i] - 'a')) % MOD;
        if (i) {
            P1 = (BASE * P1) % MOD;
        }
    }

    if (caut_bin(hash1) == 1) {
       ++sol;
    }

    for (int i = n; i < m; ++i) {
        hash1 = ((hash1 - ((prop[i - n] - 'a') * P1) % MOD + MOD) * BASE + (prop[i] - 'a')) % MOD;

        if (caut_bin(hash1) == 1) {
            ++sol;
        }
    }

    g << sol;
    return 0;
}