Cod sursa(job #2675671)

Utilizator BogdanTicuTicu Bogdan Valeriu BogdanTicu Data 22 noiembrie 2020 12:19:10
Problema Potrivirea sirurilor Scor 6
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.77 kb
#include <bits/stdc++.h>

using namespace std;

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

unsigned int MOD=197;
unsigned int B=131;

string a,b;
unsigned long long hashA,hashB,powe;
int sol;
vector<int> ans;
void make_hash()
{
	powe=1;
	for(int i=0;i<a.length();i++)
	{
		hashA=(hashA * B + a[i]) % MOD;
		powe=(powe * B) % MOD;
	}
	for(int i=0;i<a.length();i++)
		hashB=(hashB * B + b[i]) % MOD;
}

int main()
{
	in>>a>>b;
	make_hash();
	if(hashA == hashB)
	{
		sol++;
		ans.push_back(0);
	}
	for(int i=1;i<=b.size()-a.size();i++)
	{
		hashB=(hashB * B+b[i+a.size()-1]) % MOD;
		hashB=(hashB-((powe * b[i-1]) % MOD) + MOD) % MOD;
		if(hashA==hashB)
		{
			sol++;
			if(sol<=1000)
				ans.push_back(i);
		}
	}
	out<<sol<<"\n";
	for(auto x:ans)
		out<<x<<" ";
	return 0;
}