Cod sursa(job #1727358)

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

using namespace std;

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

int v[50005] , put[50];

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() {
    freopen("abc2.in" , "r" , stdin);
    freopen("abc2.out" , "w" , stdout);

    scanf("%s" , &prop);
    m = strlen(prop);

    put[0] = 1;
    for (int i = 1; i <= 50; ++i) {
        put[i] = put[i - 1] * 3;
    }
    while (scanf("%s" , &cuv) != EOF) {
        int hash1 = 0;
        n = strlen(cuv);

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

        v[++nr] = hash1;
    }

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

    int P1 = 1, hash1 = 0;
    for (int i = 0; i < n; ++i) {
         hash1 += (prop[i] - 'a') * put[i];
    }

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

    for (int i = n; i < m; ++i) {
        hash1 /= 3;
        hash1 += (prop[i] - 'a') * put[n - 1];

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

    printf("%d" , sol);
    return 0;
}