Cod sursa(job #2555340)

Utilizator CyborgSquirrelJardan Andrei CyborgSquirrel Data 23 februarie 2020 21:48:42
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.04 kb
#include <iostream>
#include <fstream>
#include <string>
#include <bitset>

using namespace std;

typedef long long lint;

lint cv(char c){
	return c-'a';
}

struct roller{
	const lint modit = 10000141, alp = 3;
	lint val = 0, rm = 1;
	void extend(char a){
		val = ((val*alp)+cv(a)) % modit;
		rm = (rm*alp) % modit;
	}
	void extend(const string &s){
		for(auto c : s){
			extend(c);
		}
	}
	void roll(char a, char b){
		val = (val*alp) - rm*cv(a) + cv(b);
		val = (val+modit*10) % modit;
	}
};
bitset<10000141> bi;

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

string s, q;
void altoire(const string &a){
	roller r;
	r.extend(a);
	bi[r.val] = true;
}

int n;
int sol = 0;
void rezolvescu(){
	roller r;
	for(int i = 0; i < n; ++i){
		r.extend(s[i]);
	}
	if(bi[r.val]){
		sol++;
	}
	for(int i = n; i < s.size(); ++i){
		r.roll(s[i-n], s[i]);
		if(bi[r.val]){
			sol++;
		}
	}
	fout << sol;
}

int main(){
	fin >> s;
	while(fin >> q){
		n = q.size();
		altoire(q);
	}
	rezolvescu();
	return 0;
}