Cod sursa(job #3125027)

Utilizator hhhhhhhAndrei Boaca hhhhhhh Data 1 mai 2023 14:10:16
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.97 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("strmatch.in");
ofstream fout("strmatch.out");
string A,B;
int z[(int)4e6+1];
void prefix(string &s)
{
    z[0]=0;
    int l=0;
    int r=0;
    for(int i=1;i<s.size();i++)
    {
        if(i<=r)
            z[i]=min(z[i-l],r-i+1);
        while(i+z[i]<s.size()&&s[z[i]]==s[i+z[i]])
            z[i]++;
        if(i+z[i]-1>r)
        {
            l=i;
            r=i+z[i]-1;
        }
    }
}
int main()
{
    ios_base::sync_with_stdio(false);
    fin.tie(0);
    fin>>A>>B;
    B=A+"#"+B;
    prefix(B);
    vector<int> sol;
    int ans=0;
    /*fout<<B<<'\n';
    for(int i=0;i<B.size();i++)
        fout<<z[i]<<' ';
    fout<<'\n';*/
    for(int i=A.size()+1;i<B.size();i++)
        if(z[i]==A.size())
    {
        ans++;
        if(sol.size()<1000)
            sol.push_back(i-A.size()-1);
    }
    fout<<ans<<'\n';
    for(int i:sol)
        fout<<i<<' ';
    return 0;
}