Cod sursa(job #534229)

Utilizator Robert29FMI Tilica Robert Robert29 Data 15 februarie 2011 15:02:40
Problema Abc2 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.09 kb
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
FILE*f=fopen("abc2.in","r");
FILE*g=fopen("abc2.out","w");
using namespace std;
long long w[50001];
char v[10000001],a[50001][20];
int N;
int cautbin(long long val){
	
	int i, step;
    for (step = 1; step <= N; step <<= 1);
    for (i = 1; step; step >>= 1)
        if (i + step <= N && w[i + step] <= val)
           i += step;
    return w[i] == val ;
}
int main(){
	int i=1;
	fscanf(f,"%s",&v);
	int x1=strlen(v);
	fscanf(f,"%s",&a[i]);
	while(a[i][0])
		fscanf(f,"%s",&a[++i]);
	int x=strlen(a[1]);
	i--;
	N=i;
	for(int j=1;j<=i;++j){
		long long z=0;
		long long t=1;
		for(int k=0;k<x;++k){
			z+=(a[j][k]-'a')*t;
			t*=3;
		}
		w[j]=z;
	}
	sort(w+1,w+i+1);
	long long z=0;
	long long t=1;
	for(int j=0;j<x;++j){
		z+=(v[j]-'a')*t;
		t*=3;	
	}
	int nr=0;
	if(cautbin(z))
		nr++;
	int d=pow(3,x-1);		
	for(register int j=x;j<x1;++j){
		z=z-(v[j-x]-'a');
		z/=3;
		z+=(v[j]-'a')*d;
		
		N=i;
		if(cautbin(z))
			nr++;
		
	}		
	fprintf(g,"%d",nr);
	
	fclose(g);
	fclose(f);
	return 0;
}