Cod sursa(job #2480847)

Utilizator PandaChanTrusca Daria PandaChan Data 26 octombrie 2019 10:38:44
Problema Potrivirea sirurilor Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.06 kb
#include <iostream>
#include <fstream>
#include <cstring>
using namespace std;
ifstream f("strmatch.in");
ofstream g("strmatch.out");
const int MAXN=2000005;
char a[MAXN], b[MAXN];
int n, m, t;
int prefix[MAXN], pozitii[MAXN];
void sir_prefix()
{
    int pi=0;
    for(int i=2; i<=n; i++)
    {
        while(a[i]!=a[pi+1] && pi!=0)
            pi=prefix[pi];
        if(a[i]==a[pi+1])
            pi++;
        prefix[i]=pi;
    }
}
int KMP()
{
    int pi=0, nr=0;
    for(int i=1; i<=m; i++)
    {
        while(b[i]!=a[pi+1] && pi!=0)
            pi=prefix[pi];
        if(b[i]==a[pi+1])
            pi++;
        if(pi==n)
        {
            nr++;
            pozitii[t++]=i-n;
        }
    }
    return nr;
}
int main()
{
    f.getline(a+1, MAXN);
    f.getline(b+1, MAXN);
    n=strlen(a+1);
    m=strlen(b+1);
    sir_prefix();
    g<<KMP()<<'\n';
    if(t<=1000)
        for(int i=0; i<t; i++)
            g<<pozitii[i]<<' ';
    else
        for(int i=0; i<=1000; i++)
            g<<pozitii[i]<<' ';
    return 0;
}