Cod sursa(job #2330217)

Utilizator triscacezarTrisca Vicol Cezar triscacezar Data 28 ianuarie 2019 08:31:54
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.75 kb
#include <bits/stdc++.h>

using namespace std;

ifstream f("strmatch.in");
ofstream g("strmatch.out");

int n,m,i,j,k,kmp[2000010];
vector<int> ans;
string A,B;

int main()
{
    f>>A>>B;
    n=A.size();
    m=B.size();
    for(i=2;i<=n;i++)
    {
        while(k&&A[k]!=A[i-1])
            k=kmp[k];
        if(A[k]==A[i-1])
            k++;
        kmp[i]=k;
    }
    k=0;int cnt=0;
    for(i=1;i<=m;i++)
    {
        while(k&&A[k]!=B[i-1])
            k=kmp[k];
        if(A[k]==B[i-1])
            k++;
        if(k==n)
            if(ans.size()!=1000)
                ans.push_back(i-n);
            else
                cnt++;
    }
    g<<ans.size()+cnt<<'\n';
    for(auto it:ans)
        g<<it<<' ';
    return 0;
}