Cod sursa(job #2555537)

Utilizator CyborgSquirrelJardan Andrei CyborgSquirrel Data 24 februarie 2020 09:39:07
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.4 kb
#include <string>
#include <bitset>
#include <vector>
#include <cmath>
#include <cstdio>
#include <cstring>

using namespace std;

typedef unsigned int uint;

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

int n, m;

const uint c1 = 666013;
const uint alp = 3;
uint rm;

vector<uint> ve[c1];
struct roller{
	uint val = 0;
	void extend(char a){
		val = (val*alp)+cv(a);
	}
	void extend(char *s){
		for(int i = 0; i < n; i++){
			extend(s[i]);
		}
	}
	void roll(char a, char b){
		val = (val - rm*cv(a))*alp + cv(b);
	}
	bool chek(){
		const vector<uint> &v = ve[val%c1];
		for(auto a : v){
			if(a == val){
				return true;
			}
		}
		return false;
	}
	void setz(){
		if(!chek()){
			ve[val%c1].push_back(val);
		}
	}
};

FILE *fin, *fout;
char s[10000041], q[41];

void altoire(){
	roller r;
	r.extend(q);
	r.setz();
}

int sol = 0;
void rezolvescu(){
	roller r;
	for(int i = 0; i < n; ++i){
		r.extend(s[i]);
	}
	if(r.chek()){
		sol++;
	}
	for(int i = n; s[i]!='\0' && s[i]!='\n'; ++i){
		r.roll(s[i-n], s[i]);
		if(r.chek()){
			sol++;
		}
	}
	printf("%d", sol);
}

int main(){
	fin = freopen("abc2.in", "r", stdin);
	fout = freopen("abc2.out", "w", stdout);
	fgets(s, 10000041, fin);
	while(fgets(q, 41, fin)){
		if(n==0){
			n = strlen(q);
			if(q[n-1] == '\n'){
				n--;
			}
		}
		altoire();
	}
	rm = pow(alp, n-1);
	rezolvescu();
	return 0;
}