Cod sursa(job #977935)

Utilizator crisbodnarCristian Bodnar crisbodnar Data 27 iulie 2013 09:53:28
Problema Potrivirea sirurilor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 0.97 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <cstring>

using namespace std;

ifstream fin("strmatch.in");
ofstream fout("strmatch.out");

const int N = 2000110;

char P[N], T[N];
int n, m, sol, urm[N];
vector <int> poz;

void Prefix()
{
    int k = -1;
    for(int q=1; q<m; q++)
    {
        while(k > 0 && P[k+1] != P[q])
            k = urm[k];
        if(P[k+1] == P[q]) k++;
        urm[q] = k;
    }
}

int main()
{
    fin.getline(P, N); fin.getline(T, N);
    m = strlen(P), n = strlen(T);
    Prefix(); int x = -1;
    for(int i=0; i<n; i++)
    {
        while(x > 0 && P[x+1] != T[i])
            x = urm[x];
        if(P[x+1] == T[i]) x++;
        if(x == m-1)
        {
            sol++;
            x = urm[x];
            if(sol <= 1000) poz.push_back(i-m+1);
        }
    }
    fout<<sol<<'\n';
    if(sol > 1000) sol = 1000;
    for(unsigned i=0; i<sol; i++)
        fout<<poz[i]<<' ';
    return 0;
}