Cod sursa(job #2307179)

Utilizator usureluflorianUsurelu Florian-Robert usureluflorian Data 23 decembrie 2018 21:59:08
Problema Potrivirea sirurilor Scor 36
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.77 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f ("strmatch.in");
ofstream g ("strmatch.out");
const int nmax=2e6+3;
char s1[nmax],s2[nmax];
int n1,n2,k,ant[nmax];
vector <int> sol;
int main()
{
    f>>s1;
    f>>s2;
    n1=strlen(s1);
    n2=strlen(s2);
    k=0;
    for(int i=2;i<=n1;++i)
    {
        while(k&&s1[i]!=s1[k+1]) k=ant[k];
        if(s1[i]==s1[k+1]) ++k;
        ant[i]=k;
    }
    k=0;
    for(int i=1;i<=n2;++i)
    {
        while(k&&s1[k+1]!=s2[i]) k=ant[k];
        if(s1[k+1]==s2[i]) ++k;
        if(k==n1-1)
        {
            sol.push_back(i-k);
            if(sol.size()==1000) break;
            k=ant[k];
        }
    }
    g<<sol.size()<<'\n';
    for(int i=0;i<sol.size();++i) g<<sol[i]<<' ';
    return 0;
}