Cod sursa(job #102524)

Utilizator peanutzAndrei Homorodean peanutz Data 14 noiembrie 2007 14:51:35
Problema Abc2 Scor 0
Compilator cpp Status done
Runda Happy Coding 2007 Marime 1.18 kb
#include <stdio.h>
#include <vector>
#include <set>

using namespace std;

#define pb push_back
#define MOD 301001
#define NMAX 10000010

inline long long f(int n)
{
	if(n == 0) return 1;
	long long p = 1;
	for(int i = 1; i < n; ++i)
		p = (long long)p * 3;
	return p;
}

char s[NMAX], c[21];
int n;
long long res;
set<long long> h[MOD];

void read()
{
	scanf("%s", c);
	if(!n)
		n = strlen(c);
	long long x = 0;
	for(int i = 0; i < n; ++i)
		x = (long long)x*3 + (c[i]-'a');
	h[x % MOD].insert(x);
    //printf("x %d\n", x);
}

int main()
{
	freopen("abc2.in", "r", stdin);
	freopen("abc2.out", "w", stdout);

	scanf("%s", s);
	while(!feof(stdin))
		read();
	long long t = 0, p = f(n);
	for(int i = 0; i < n; ++i)
		t = (long long)t*3 + (s[i]-'a');

	set<long long> :: iterator it;
	long long until = strlen(s)-n, aux;
	for(int i = 0; i <= until; ++i)
	{
		aux = t % MOD;//  printf("%d\n", t);
		
		if(h[aux].find(t) != h[aux].end())
			++res;

       		t = (long long)t - p*(s[i]-'a');
       		t = (long long)t*3 + (s[i+n]-'a');
	}
	printf("%lld\n", res);
	return 0;
}