Cod sursa(job #2807008)

Utilizator victorzarzuZarzu Victor victorzarzu Data 23 noiembrie 2021 11:53:33
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.95 kb
#include <bits/stdc++.h>
#define oo 0x3f3f3f3f 

using namespace std;
ifstream f("strmatch.in");
ofstream g("strmatch.out");
int len_a, len_b, number;
int pi[2000001];
string a, b;
vector<int> result;

void read()
{
  f>>a>>b;
  len_a = a.length(), len_b = b.length();
  a.insert(0, " ");
  b.insert(0, " ");
} 

void precompute()
{
  int q = 0;
  for(int i = 2;i <= len_a;++i)
    {
      while(q && a[q + 1] != a[i])
        q = pi[q];
      if(a[q + 1] == a[i])
        ++q;
      pi[i] = q;
    }
}

void solve()
{
  precompute();
  int q = 0;
  for(int i = 1;i <= len_b;++i)
  {
    while(q && a[q + 1] != b[i])
      q = pi[q];
    if(a[q + 1] == b[i])
      ++q;
    if(q == len_a) 
    {
      q = pi[q];
      ++number;
      if(number <= 1000)
        result.push_back(i - len_a);
    }
  }
  g<<number<<'\n';
  for(auto it : result)
    g<<it<<" ";

}

int main()
{
  read();
  solve();
  return 0;
}