Cod sursa(job #1059639)

Utilizator kitzTimofte Bogdan kitz Data 16 decembrie 2013 20:41:25
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.2 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);
    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]; A[0] = ' ';
    for (i = N; i; --i) B[i] = B[i-1]; B[0] = ' ';  
 //M = strlen(A)-1; N = strlen(B)-1;
 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 <= minim(n, 1000); ++i) g<<pos[i]<<" ";
 g<<"\n"; return 0;
}