結果

問題 No.2102 [Cherry Alpha *] Conditional Reflection
ユーザー 👑 emthrm
提出日時 2022-10-14 22:25:15
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,189 ms / 3,000 ms
コード長 3,503 bytes
コンパイル時間 2,221 ms
コンパイル使用メモリ 211,604 KB
最終ジャッジ日時 2025-02-08 04:20:48
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 70
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define _USE_MATH_DEFINES
#include <bits/stdc++.h>
using namespace std;
#define FOR(i,m,n) for(int i=(m);i<(n);++i)
#define REP(i,n) FOR(i,0,n)
#define ALL(v) (v).begin(),(v).end()
using ll = long long;
constexpr int INF = 0x3f3f3f3f;
constexpr long long LINF = 0x3f3f3f3f3f3f3f3fLL;
constexpr double EPS = 1e-8;
constexpr int MOD1 = 998244353;
constexpr int MOD2 = 1000000007;
constexpr int DY4[]{1, 0, -1, 0}, DX4[]{0, -1, 0, 1};
constexpr int DY8[]{1, 1, 0, -1, -1, -1, 0, 1};
constexpr int DX8[]{0, -1, -1, -1, 0, 1, 1, 1};
template <typename T, typename U>
inline bool chmax(T& a, U b) { return a < b ? (a = b, true) : false; }
template <typename T, typename U>
inline bool chmin(T& a, U b) { return a > b ? (a = b, true) : false; }
struct IOSetup {
  IOSetup() {
    std::cin.tie(nullptr);
    std::ios_base::sync_with_stdio(false);
    std::cout << fixed << setprecision(20);
  }
} iosetup;

template <typename T = std::string>
struct RollingHash {
  T s;

  explicit RollingHash(const T& s, const int base = 10007,
                       const int mod = 1000000007)
      : base(base), mod(mod), hash({0}), power({1}) {
    const int n = s.size();
    this->s.reserve(n);
    hash.reserve(n + 1);
    power.reserve(n + 1);
    add(s);
  }

  long long get(const int left, const int right) const {
    const long long res =
        hash[right] - hash[left] * power[right - left] % mod;
    return res < 0 ? res + mod : res;
  }

  void add(const T& t) {
    for (const auto c : t) {
      s.push_back(c);
      const int hash_nxt = (hash.back() * base % mod + c) % mod;
      hash.emplace_back(hash_nxt);
      const int power_nxt = power.back() * base % mod;
      power.emplace_back(power_nxt);
    }
  }

  int longest_common_prefix(const int i, const int j) const {
    const int n = s.size();
    int lb = 0, ub = n + 1 - std::max(i, j);
    while (ub - lb > 1) {
      const int mid = (lb + ub) >> 1;
      (get(i, i + mid) == get(j, j + mid) ? lb : ub) = mid;
    }
    return lb;
  }

  template <typename U>
  int longest_common_prefix(const RollingHash<U>& t,
                            const int i, const int j) const {
    int lb = 0;
    int ub = std::min(static_cast<int>(s.size()) - i,
                      static_cast<int>(t.s.size()) - j)
             + 1;
    while (ub - lb > 1) {
      const int mid = (lb + ub) >> 1;
      (get(i, i + mid) == t.get(j, j + mid) ? lb : ub) = mid;
    }
    return lb;
  }

  std::vector<long long> power;
 private:
  const int base, mod;
  std::vector<long long> hash;
};

int main() {
  int n; cin >> n;
  map<int, set<pair<int, int>>> hashes;
  while (n--) {
    string s; cin >> s;
    RollingHash rh1(s, 10007, MOD1), rh2(s, 10007, MOD2);
    const int l = s.length(), hash1 = rh1.get(0, l), hash2 = rh2.get(0, l);
    cout << (hashes[l].count({hash1, hash2}) ? "Yes\n" : "No\n");
    hashes[l].emplace(hash1, hash2);
    FOR(i, 1, l) {
      ll nhash1 = hash1, nhash2 = hash2;
      nhash1 -= 1LL * rh1.power[l - 1 - (i - 1)] * s[i - 1];
      nhash1 -= 1LL * rh1.power[l - 1 - i] * s[i];
      nhash1 += 1LL * rh1.power[l - 1 - (i - 1)] * s[i];
      nhash1 += 1LL * rh1.power[l - 1 - i] * s[i - 1];
      nhash2 -= 1LL * rh2.power[l - 1 - (i - 1)] * s[i - 1];
      nhash2 -= 1LL * rh2.power[l - 1 - i] * s[i];
      nhash2 += 1LL * rh2.power[l - 1 - (i - 1)] * s[i];
      nhash2 += 1LL * rh2.power[l - 1 - i] * s[i - 1];
      hashes[l].emplace((nhash1 % MOD1 + MOD1) % MOD1, (nhash2 % MOD2 + MOD2) % MOD2);
    }
  }
  return 0;
}
0