Cod sursa(job #369740)

Utilizator titusuTitus C titusu Data 29 noiembrie 2009 14:24:50
Problema Potrivirea sirurilor Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.8 kb
using namespace std;
#include <cstdio>
#define MAX 2000010
#define nrLitere (26+26+10)
#define nrPrim 100021
char s[MAX], p[MAX];
int ns,np;
FILE * fin= fopen("strmatch.in","r");
FILE * fout=fopen("strmatch.out","w");

void read();
void solve();
void write();
int ord(char c);
int main(){
		read();
		/*
		for(int i=1;i<=np;i++)
			cout<<p[i];
		cout<<endl;
		for(int i =1;i<=ns;i++)
			cout<<s[i];
		cout<<endl;
		*/
		solve();
		write();
		/*
		int ppp=0;
		for(int i=232842, j=1;j<=np;j++)
			if(s[i+j]==p[j])
				ppp++,cout<<j<<" ";
		cout<<endl<<ppp<<endl;
		*/
		return 0;
	}

int isOk(char c){
	if(c>='A' && c<='Z')
		return 1;
	if(c>='a' && c<='z')
		return 1;
	if(c>='0' && c<='9')
		return 1;
	return 0;
}

void read(){
	ns=np=0;
	fscanf(fin,"%s%s",p,s);
	for( ; s[ns] ; ++ns);
	for( ; p[np] ; ++np);
	int i;
	for( i=ns; i ; --i)
		s[i]=s[i-1];
	for( i=np; i ; --i)
		p[i]=p[i-1];
}

int nrpoz, nrpp, poz[1010];

void solve(){
	int dlam_1=1; //(d=nrLitere)^(np-1)
	for(int i=1;i<np;++i)
		dlam_1 =(dlam_1 *nrLitere) % nrPrim;
	int hp=0 ; //hashul lui p
	for(int i=1;i<=np;i++)
		hp = (hp * nrLitere + ord(p[i])) % nrPrim;
	//cout<<"hp = "<<hp<<endl;
	int hs = 0; // hashul lui s1s2...snp, se modifica mai tarziu
	for(int i=1;i<=np;i++)
		hs = (hs * nrLitere + ord(s[i])) % nrPrim;
	for(int i=1;i<=ns-np+1;++i){
		//cout<<i<<" "<<s[i]<<" "<<hs<<endl;
		//verific protrivirea
		if(hp ==hs && s[i+np-1]==p[np] && s[i]==p[1]){
			nrpoz++;
			if(nrpoz<=1000)
				poz[++nrpp]=i-1;
		}
		hs = (hs + nrPrim-(ord(s[i])*dlam_1)%nrPrim)%nrPrim;
		hs = (hs*nrLitere+ord(s[i+np])) % nrPrim;
	}
}

void write(){
	fprintf(fout,"%d\n",nrpoz);
//	fout<<nrpoz<<endl;
	for(int i=1;i<=nrpp;++i)
//		fout<<poz[i]<<" ";
		fprintf(fout,"%d ",poz[i]);
}

int ord(char c){
	if(c<='9')
		return c-'0';
	if(c<='Z')
		return 10+c-'A';
	return 10+26+c-'a';
}