Cod sursa(job #2199833)

Utilizator PushkinPetolea Cosmin Pushkin Data 29 aprilie 2018 11:43:25
Problema Potrivirea sirurilor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 0.87 kb
#include <iostream>
#include <bits/stdc++.h>
#include <fstream>
using namespace std;
ifstream f("strmatch.in");
ofstream g("strmatch.out");
string a, b;
vector<int> p, res;
int n, m;
void prefix()
{
    int k=0, i;
    p.resize(m+3);
    p[1]=0;
    for(i=2;i<=m;i++)
    {
        while(k&&b[i]!=b[k+1])
            k=p[i];
        if(b[i]==b[k+1])
            k++;
        p[i]=k;
    }
}
void KMP()
{
    int k=0, i;
    n=a.size();
    m=b.size();
    a=' '+a;
    b=' '+b;
    prefix();
    for(i=1;i<=n;i++)
    {
        while(k&&a[i]!=b[k+1])
            k=p[k];
        if(a[i]==b[k+1])
            k++;
        if(k==m)
            res.push_back(i-k);
    }
}
int main()
{
    getline(f, b);
    getline(f, a);
    KMP();
    g<<res.size()<<'\n';
    for(auto x:res)
        g<<x<<' ';
    f.close();
    g.close();
    return 0;
}