Cod sursa(job #1059643)

Utilizator kitzTimofte Bogdan kitz Data 16 decembrie 2013 20:51:01
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.41 kb
//KMP_wikipedia.cpp : 
//#include "stdafx.h"
#include <fstream>
#include <cstring>
using namespace std;
//#define minim(a, b) ((a < b) ? a : b)
#define NMax 2000005
ifstream f ("strmatch.in"); ofstream g ("strmatch.out");
int M, N;
char A[NMax], B[NMax];
int pi[NMax], pos[1024];
void prefix()
{int i, q = 0;
 for (i = 2, pi[1] = 0; i <= M; ++i){
      while (q && A[q+1] != A[i]) q = pi[q];
      if (A[q+1] == A[i]) ++q; pi[i] = q;
    }
}
 
int main()
{int i, q = -1, n = 0;
 f>>A>>B;
 /*for (; (A[M] >= 'A' && A[M] <= 'Z') || (A[M] >= 'a' && A[M] <= 'z') 
            || (A[M] >= '0' && A[M] <= '9'); ++M);*/
 while((A[M] >= 'A' && A[M] <= 'Z') || (A[M] >= 'a' && A[M] <= 'z') 
            || (A[M] >= '0' && A[M] <= '9') ) ++M;
 while((B[N] >= 'A' && B[N] <= 'Z') || (B[N] >= 'a' && B[N] <= 'z')
            || (B[N] >= '0' && B[N] <= '9')) ++N;
 /*for (; (B[N] >= 'A' && B[N] <= 'Z') || (B[N] >= 'a' && B[N] <= 'z')
            || (B[N] >= '0' && B[N] <= '9'); ++N);*/
 for (i = M; i; --i) 
	 A[i] = A[i-1]; 
 for (i = N; i; --i) 
	 B[i] = B[i-1]; 
 A[0] = ' '; B[0] = ' ';  
 prefix();
 for (i = 1; i <= N; ++i)
    {while (q && A[q+1] != B[i]) q = pi[q];
     if (A[q+1] == B[i]) ++q;
     if (q == M)
        {q = pi[M];
         ++n;
         if (n <= 1000) pos[n] = i-M;    
        }   
    }       
 g<<n<<"\n";
 for (i = 1; i <= min(n, 1000); ++i) g<<pos[i]<<" ";
 g<<"\n"; return 0;
}