Cod sursa(job #497219)

Utilizator ZethpixZethpix Zethpix Data 1 noiembrie 2010 21:08:48
Problema Abc2 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.48 kb
#include <stdio.h>
#include <string.h>

int n,m,nr,sol,pow,i,M,sol1,pow1,M1;

struct hash
{
	int nod,nod1;
	hash *link;
}*H[45000];

char str[10000005],s[25];

void add(int x,int x1)
{
	hash *p;
	p=new hash;
	p->nod=x;
	p->nod1=x1;
	p->link=H[x%M];
	H[x%M]=p;
}

int src(int x,int x1)
{
	hash *p;
	p=H[x%M];
	while(p!=NULL)
	{
		if(p->nod==x&&p->nod1==x1) return 1;
		p=p->link;
	}
	return 0;
}

int main()
{
	freopen("abc2.in","r",stdin);
	freopen("abc2.out","w",stdout);

	fgets(str,10000005,stdin);
	M=44027;
	M1=43997;
	n=strlen(str)-1;
	if(str[n]=='\n') n--;
	fgets(s,25,stdin);
	m=strlen(s)-1;
	if(s[m]=='\n') m--;
	sol=0;
	sol1=0;
	for(i=0;i<=m;i++)
	{
		sol=((sol*4)%M+s[i]-'a')%M;
		sol1=((sol1*4)%M1+s[i]-'a')%M1;
	}
	if(src(sol,sol1)==0) add(sol,sol1);
	while(!feof(stdin))
	{
		fgets(s,25,stdin);
		if(!feof(stdin))
		{
			sol=0;
			sol1=0;
			for(i=0;i<=m;i++)
			{
				sol=((sol*4)%M+s[i]-'a')%M;
				sol1=((sol1*4)%M1+s[i]-'a')%M1;
			}
			add(sol,sol1);
		}
	}

	sol=0;
	pow=1;
	sol1=0;
	pow1=0;
	nr=0;
	for(i=1;i<=m;i++)
	{
		pow=(pow*4)%M;
		pow1=(pow1*4)%M1;
	}
	for(i=0;i<=m;i++)
	{
		sol=((sol*4)%M+str[i]-'a')%M;
		sol1=((sol1*4)%M1+str[i]-'a')%M1;
	}
	if(src(sol,sol1)) nr++;
	for(i=m+1;i<=n;i++)
	{
		sol=(((sol+M-(pow*(str[i-m-1]-'a'))%M)%M)*4+(str[i]-'a'))%M;
		sol1=(((sol1+M1-(pow1*(str[i-m-1]-'a'))%M1)%M1)*4+(str[i]-'a'))%M1;
		if(src(sol,sol1)) nr++;
	}
	printf("%d\n",nr);

	return 0;
}