Submission #46595821

---

Source Code Expand

#include <bits/stdc++.h>
#ifdef LOCAL
#include "debug.h"
#else
#define debug(...)
#endif
using int64 = long long;
using uint = unsigned int;
using uint64 = unsigned long long;
bool ckmin(auto& a, auto b) { return b < a ? a = b, 1 : 0; }
bool ckmax(auto& a, auto b) { return b > a ? a = b, 1 : 0; }
using namespace std;

int CP(string& a, string& b) {
  int i = 0, j = 0, res = 0;
  while (i < (int)a.length() && j < (int)b.length()) {
    if (a[i] == b[j]) res++;
    else break;
    i++, j++;
  }
  return res;
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);

  int n;
  string s0;
  cin >> n >> s0;
  string s0r = s0;
  reverse(s0r.begin(), s0r.end());
  vector<int> ans;
  for (int i = 0; i < n; i++) {
    string s;
    cin >> s;
    if (s == s0) {
      ans.push_back(i);
      continue;
    }
    int l = CP(s0, s);
    reverse(s.begin(), s.end());
    int r = CP(s0r, s);
    int t = l + r, x = s0.length(), y = s.length();
    debug(l, r, t, x, y);
    if (abs(x - y) <= 1 && t >= x - 1 && t >= y - 1) ans.push_back(i);
  }
  cout << ans.size() << '\n';
  for (int x : ans) cout << x + 1 << ' ';
  cout << '\n';

  return 0;
}

Submission Info

Judge Result