Cod sursa(job #3225424)

Utilizator hhhhhhhAndrei Boaca hhhhhhh Data 17 aprilie 2024 16:16:00
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.89 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("strmatch.in");
ofstream fout("strmatch.out");
string a,b,s;
vector<int> sol;
int ans=0;
int z[4000005];
void z_function()
{
    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)
        {
            r=i+z[i]-1;
            l=i;
        }
    }
}
int main()
{
    ios_base::sync_with_stdio(false);
    fin.tie(0);
    fin>>a>>b;
    s=a+"#"+b;
    z_function();
    int ans=0;
    for(int i=0;i<s.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;
}