Cod sursa(job #2176369)

Utilizator TherevengerkingSurani Adrian Therevengerking Data 17 martie 2018 00:00:57
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.96 kb
#include <iostream>
#include <vector>
#include <fstream>
#include <string.h>
using namespace std;
ifstream fin("strmatch.in");
ofstream fout("strmatch.out");
const int Nmax = 2000000 + 5;

char C[Nmax], P[Nmax];
int p[Nmax];
#define pb push_back
vector<int>ans;
int main()
{
    fin >> P + 1 >> C + 1;
    int N = strlen(C + 1), M = strlen(P + 1);
    int k = 0; p[1] = 0;p[0] = 0;
    for(int i = 2; i <= M; ++i)
    {
        while(k > 0 && P[k + 1] != P[i])
            k = p[k];
        if(P[i] == P[k + 1])
            ++k;
        p[i] = k;
    }
    k = 0;
    for(int i = 1; i <= N; ++i)
    {
        while(k > 0 && P[k + 1] != C[i])
            k = p[k];
        if(C[i] == P[k + 1])
            ++k;
        if(k == M)
        {
            k = p[k];
            ans.pb(i - M);
        }
    }
    fout << ans.size() << '\n';
    for(int i = 0; i < min((int)ans.size(), 1000); ++i)
        fout << ans[i] << ' ';
    return 0;
}