Cod sursa(job #2569881)

Utilizator hudisteanumihaelaHudisteanu Mihaela hudisteanumihaela Data 4 martie 2020 14:03:37
Problema Potrivirea sirurilor Scor 32
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.83 kb
#include <bits/stdc++.h>
#define Nmax 2000005

using namespace std;
ifstream fin ("strmatch.in");
ofstream fout ("strmatch.out");
char a[Nmax], b[Nmax];
int pi[Nmax], n, m, k;
vector <int> sol;
int main()
{
    fin.getline(a+1, Nmax);
    fin.getline(b+1, Nmax);
    a[0]='1';
    b[0]='1';
    n=strlen(a)-1;
    m=strlen(b)-1;
    for(int i=2;i<=n;i++)
    {
        while(k>0 && a[k+1]!=a[i])
            k=pi[k];
        if(a[k+1]==a[i])
            k++;
        pi[i]=k;
    }
    for(int i=1;i<=m;i++)
    {
        while(k>0 && a[k+1]!=b[i])
            k=pi[k];
        if(a[k+1]==b[i])
            k++;
        if(k==n)
            sol.push_back(i-n);
        if(sol.size()==1000)
            break;
    }
    fout << sol.size() << '\n';
    for(auto it:sol)
        fout << it << ' ';
    return 0;
}