Cod sursa(job #2135809)

Utilizator dumitru123Patularu Mihai dumitru123 Data 19 februarie 2018 11:37:33
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.88 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("strmatch.in");
ofstream g("strmatch.out");
#define nmax 2000003
char a[nmax],b[nmax];
int prefix[nmax];
vector < int > sol;
void calcul(char a[])
{
    int x=strlen(a);
    int k=0;
    for(int i=1; i<x; i++)
    {
        while(a[i]!=a[k] && k>0)
            k=prefix[k-1];
        if(a[i]==a[k])
            k++;
        prefix[i]=k;
    }
}
void KMP(char a[], char b[])
{
    int k=0;
    int x=strlen(b);
    int y=strlen(a);
    for(int i=0; i<x; i++)
    {
        while(a[k]!=b[i] && k>0)
            k=prefix[k-1];
        if(a[k]==b[i])
            k++;
        if(k==y)
            sol.push_back(i-k+1);
    }
}
int main()
{
    f>>a>>b;
    calcul(a);
    KMP(a,b);
    g<<sol.size()<<'\n';
    for(int i=0; i<min((int)sol.size(),1000); i++)
        g<<sol.at(i)<<' ';
    return 0;
}