Cod sursa(job #1660736)

Utilizator ArkinyStoica Alex Arkiny Data 23 martie 2016 13:15:20
Problema Abc2 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.04 kb
#include<fstream>
#include<string>
#include<vector>
#include<string.h>
#include<iostream>
using namespace std;

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



char S[10000000];
char s1[22];
long long pow1[20];

vector<long long> hash1[500000];
#define MOD 66537
int main()
{
	in >> S;
	long long p = 1;
	for (int i = 0; i < 20; ++i)
		pow1[i] = p, p *= 3;
	int ok = 0 , size = 0;
	while (in >> s1)
	{
		int i = 0;
		long long s = 0;
		while (s1[i] != '\0')
			s += (s1[i] - 'a')*pow1[i], ++i;
		
		hash1[s%MOD].push_back(s);
		if (!ok)
			size = strlen(s1);
	}

	int i = 0,nr=0;
	long long s = 0;
	for (; i < size; ++i)
		s += (S[i] - 'a')*pow1[i];

	int key = s%MOD;

	for (int i = 0; i < hash1[key].size(); ++i)
		if (hash1[key][i] == s)
		{
			++nr;
			break;
		}

	while (S[i] != '\0')
	{
		s /= 3;
		s += (S[i]-'a')*pow1[size-1];
		key = s%MOD;
		for (int i = 0; i < hash1[key].size(); ++i)
			if (hash1[key][i] == s)
			{
				++nr;
				break;
			}
		++i;
	}
	
	out << nr;
	
	
	return 0;
}