Cod sursa(job #3033421)

Utilizator dorufDoru Floare doruf Data 23 martie 2023 20:59:29
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.86 kb
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define eb emplace_back

const string TASK("strmatch");

ifstream fin(TASK + ".in");
ofstream fout(TASK + ".out");

vector<int> Z(string s) {
  int n = (int)s.length();
  vector<int> z(n);
  for (int i = 1, l = 0, r = 0; i < n; ++i) {
    if (i <= r)
      z[i] = min(r - i + 1, z[i - l]);
    while (i + z[i] < n && s[z[i]] == s[i + z[i]])
      ++z[i];
    if (i + z[i] - 1 > r)
      l = i, r = i + z[i] - 1;
  }
  return z;
}

int main() {
  string p, s;
  fin >> p >> s;
  s = p + "$" + s;
  vector<int> z = Z(s);
  vector<int> ans;
  int sz = 0;
  for (int i = p.size() + 1; i < s.size(); ++i)
    if (z[i] == p.size()) {
      if (ans.size() < 1000)
      ans.eb(i - p.size() - 1);
      ++sz;
    }

  fout << sz << '\n';
  for (auto i : ans) fout << i << ' ';
}