Cod sursa(job #2296757)

Utilizator livliviLivia Magureanu livlivi Data 5 decembrie 2018 00:03:10
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.83 kb
#include <fstream>
#include <unordered_map>
#include <string>
#define M 1000000007
using namespace std;

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

string s, x;
unordered_map<int, bool> dictionar;

int main(){
	cin >> s;

	while(cin >> x){
		long long cod = 0;
		for(int i = 0; i < x.size(); i++)
			cod = cod * 3 + x[i] - 'a';

		// cout << cod << endl;

		dictionar[(int)(cod % M)] = true;
	}

	int n = x.size();
	long long p3lan = 1;

	for(int i = 1; i < n; i++)
		p3lan *= 3;

	long long cod = 0;
	int ans = 0;
	for(int i = 0, j = 0; i <= s.size() - n; i++){
		while(j < s.size() && j < i + n){
			cod = cod * 3 + s[j] - 'a';
			// cout << cod << endl;
			j++;
		}

		if (dictionar.find((int)(cod % M)) != dictionar.end()) 
			ans ++;

		// cout << cod << endl;
		cod -= p3lan * (s[i] - 'a');
	}

	cout << ans << endl;
	return 0;
}