結果
問題 | No.2102 [Cherry Alpha *] Conditional Reflection |
ユーザー |
|
提出日時 | 2022-10-14 22:03:39 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 727 ms / 3,000 ms |
コード長 | 24,304 bytes |
コンパイル時間 | 3,771 ms |
コンパイル使用メモリ | 324,924 KB |
最終ジャッジ日時 | 2025-02-08 03:57:13 |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 70 |
ソースコード
#include <bits/stdc++.h>#ifdef _MSC_VER# include <intrin.h>#else# include <x86intrin.h>#endif#include <limits>#include <type_traits>namespace suisen {// ! utilitytemplate <typename ...Types>using constraints_t = std::enable_if_t<std::conjunction_v<Types...>, std::nullptr_t>;template <bool cond_v, typename Then, typename OrElse>constexpr decltype(auto) constexpr_if(Then&& then, OrElse&& or_else) {if constexpr (cond_v) {return std::forward<Then>(then);} else {return std::forward<OrElse>(or_else);}}// ! functiontemplate <typename ReturnType, typename Callable, typename ...Args>using is_same_as_invoke_result = std::is_same<std::invoke_result_t<Callable, Args...>, ReturnType>;template <typename F, typename T>using is_uni_op = is_same_as_invoke_result<T, F, T>;template <typename F, typename T>using is_bin_op = is_same_as_invoke_result<T, F, T, T>;template <typename Comparator, typename T>using is_comparator = std::is_same<std::invoke_result_t<Comparator, T, T>, bool>;// ! integraltemplate <typename T, typename = constraints_t<std::is_integral<T>>>constexpr int bit_num = std::numeric_limits<std::make_unsigned_t<T>>::digits;template <typename T, unsigned int n>struct is_nbit { static constexpr bool value = bit_num<T> == n; };template <typename T, unsigned int n>static constexpr bool is_nbit_v = is_nbit<T, n>::value;// ?template <typename T>struct safely_multipliable {};template <>struct safely_multipliable<int> { using type = long long; };template <>struct safely_multipliable<long long> { using type = __int128_t; };template <>struct safely_multipliable<unsigned int> { using type = unsigned long long; };template <>struct safely_multipliable<unsigned long int> { using type = __uint128_t; };template <>struct safely_multipliable<unsigned long long> { using type = __uint128_t; };template <>struct safely_multipliable<float> { using type = float; };template <>struct safely_multipliable<double> { using type = double; };template <>struct safely_multipliable<long double> { using type = long double; };template <typename T>using safely_multipliable_t = typename safely_multipliable<T>::type;template <typename T, typename = void>struct rec_value_type {using type = T;};template <typename T>struct rec_value_type<T, std::void_t<typename T::value_type>> {using type = typename rec_value_type<typename T::value_type>::type;};template <typename T>using rec_value_type_t = typename rec_value_type<T>::type;} // namespace suisen// ! type aliasesusing i128 = __int128_t;using u128 = __uint128_t;template <typename T>using pq_greater = std::priority_queue<T, std::vector<T>, std::greater<T>>;template <typename T, typename U>using umap = std::unordered_map<T, U>;// ! macros (capital: internal macro)#define OVERLOAD2(_1,_2,name,...) name#define OVERLOAD3(_1,_2,_3,name,...) name#define OVERLOAD4(_1,_2,_3,_4,name,...) name#define REP4(i,l,r,s) for(std::remove_reference_t<std::remove_const_t<decltype(r)>>i=(l);i<(r);i+=(s))#define REP3(i,l,r) REP4(i,l,r,1)#define REP2(i,n) REP3(i,0,n)#define REPINF3(i,l,s) for(std::remove_reference_t<std::remove_const_t<decltype(l)>>i=(l);;i+=(s))#define REPINF2(i,l) REPINF3(i,l,1)#define REPINF1(i) REPINF2(i,0)#define RREP4(i,l,r,s) for(std::remove_reference_t<std::remove_const_t<decltype(r)>>i=(l)+fld((r)-(l)-1,s)*(s);i>=(l);i-=(s))#define RREP3(i,l,r) RREP4(i,l,r,1)#define RREP2(i,n) RREP3(i,0,n)#define rep(...) OVERLOAD4(__VA_ARGS__, REP4 , REP3 , REP2 )(__VA_ARGS__)#define rrep(...) OVERLOAD4(__VA_ARGS__, RREP4 , RREP3 , RREP2 )(__VA_ARGS__)#define repinf(...) OVERLOAD3(__VA_ARGS__, REPINF3, REPINF2, REPINF1)(__VA_ARGS__)#define CAT_I(a, b) a##b#define CAT(a, b) CAT_I(a, b)#define UNIQVAR(tag) CAT(tag, __LINE__)#define loop(n) for (std::remove_reference_t<std::remove_const_t<decltype(n)>> UNIQVAR(loop_variable) = n; UNIQVAR(loop_variable) --> 0;)#define all(iterable) std::begin(iterable), std::end(iterable)#define input(type, ...) type __VA_ARGS__; read(__VA_ARGS__)#ifdef LOCAL# define debug(...) debug_internal(#__VA_ARGS__, __VA_ARGS__)template <class T, class... Args>void debug_internal(const char* s, T&& first, Args&&... args) {constexpr const char* prefix = "[\033[32mDEBUG\033[m] ";constexpr const char* open_brakets = sizeof...(args) == 0 ? "" : "(";constexpr const char* close_brakets = sizeof...(args) == 0 ? "" : ")";std::cerr << prefix << open_brakets << s << close_brakets << ": " << open_brakets << std::forward<T>(first);((std::cerr << ", " << std::forward<Args>(args)), ...);std::cerr << close_brakets << "\n";}#else# define debug(...) void(0)#endif// ! I/O utilities// __int128_tstd::ostream& operator<<(std::ostream& dest, __int128_t value) {std::ostream::sentry s(dest);if (s) {__uint128_t tmp = value < 0 ? -value : value;char buffer[128];char* d = std::end(buffer);do {--d;*d = "0123456789"[tmp % 10];tmp /= 10;} while (tmp != 0);if (value < 0) {--d;*d = '-';}int len = std::end(buffer) - d;if (dest.rdbuf()->sputn(d, len) != len) {dest.setstate(std::ios_base::badbit);}}return dest;}// __uint128_tstd::ostream& operator<<(std::ostream& dest, __uint128_t value) {std::ostream::sentry s(dest);if (s) {char buffer[128];char* d = std::end(buffer);do {--d;*d = "0123456789"[value % 10];value /= 10;} while (value != 0);int len = std::end(buffer) - d;if (dest.rdbuf()->sputn(d, len) != len) {dest.setstate(std::ios_base::badbit);}}return dest;}// pairtemplate <typename T, typename U>std::ostream& operator<<(std::ostream& out, const std::pair<T, U>& a) {return out << a.first << ' ' << a.second;}// tupletemplate <unsigned int N = 0, typename ...Args>std::ostream& operator<<(std::ostream& out, const std::tuple<Args...>& a) {if constexpr (N >= std::tuple_size_v<std::tuple<Args...>>) {return out;} else {out << std::get<N>(a);if constexpr (N + 1 < std::tuple_size_v<std::tuple<Args...>>) {out << ' ';}return operator<<<N + 1>(out, a);}}// vectortemplate <typename T>std::ostream& operator<<(std::ostream& out, const std::vector<T>& a) {for (auto it = a.begin(); it != a.end();) {out << *it;if (++it != a.end()) out << ' ';}return out;}// arraytemplate <typename T, size_t N>std::ostream& operator<<(std::ostream& out, const std::array<T, N>& a) {for (auto it = a.begin(); it != a.end();) {out << *it;if (++it != a.end()) out << ' ';}return out;}inline void print() { std::cout << '\n'; }template <typename Head, typename... Tail>inline void print(const Head& head, const Tail &...tails) {std::cout << head;if (sizeof...(tails)) std::cout << ' ';print(tails...);}template <typename Iterable>auto print_all(const Iterable& v, std::string sep = " ", std::string end = "\n") -> decltype(std::cout << *v.begin(), void()) {for (auto it = v.begin(); it != v.end();) {std::cout << *it;if (++it != v.end()) std::cout << sep;}std::cout << end;}__int128_t parse_i128(std::string& s) {__int128_t ret = 0;for (int i = 0; i < int(s.size()); i++) if ('0' <= s[i] and s[i] <= '9') ret = 10 * ret + s[i] - '0';if (s[0] == '-') ret = -ret;return ret;}__uint128_t parse_u128(std::string& s) {__uint128_t ret = 0;for (int i = 0; i < int(s.size()); i++) if ('0' <= s[i] and s[i] <= '9') ret = 10 * ret + s[i] - '0';return ret;}// __int128_tstd::istream& operator>>(std::istream& in, __int128_t& v) {std::string s;in >> s;v = parse_i128(s);return in;}// __uint128_tstd::istream& operator>>(std::istream& in, __uint128_t& v) {std::string s;in >> s;v = parse_u128(s);return in;}// pairtemplate <typename T, typename U>std::istream& operator>>(std::istream& in, std::pair<T, U>& a) {return in >> a.first >> a.second;}// tupletemplate <unsigned int N = 0, typename ...Args>std::istream& operator>>(std::istream& in, std::tuple<Args...>& a) {if constexpr (N >= std::tuple_size_v<std::tuple<Args...>>) {return in;} else {return operator>><N + 1>(in >> std::get<N>(a), a);}}// vectortemplate <typename T>std::istream& operator>>(std::istream& in, std::vector<T>& a) {for (auto it = a.begin(); it != a.end(); ++it) in >> *it;return in;}// arraytemplate <typename T, size_t N>std::istream& operator>>(std::istream& in, std::array<T, N>& a) {for (auto it = a.begin(); it != a.end(); ++it) in >> *it;return in;}template <typename ...Args>void read(Args &...args) {(std::cin >> ... >> args);}// ! integral utilities// Returns pow(-1, n)template <typename T>constexpr inline int pow_m1(T n) {return -(n & 1) | 1;}// Returns pow(-1, n)template <>constexpr inline int pow_m1<bool>(bool n) {return -int(n) | 1;}// Returns floor(x / y)template <typename T>constexpr inline T fld(const T x, const T y) {return (x ^ y) >= 0 ? x / y : (x - (y + pow_m1(y >= 0))) / y;}template <typename T>constexpr inline T cld(const T x, const T y) {return (x ^ y) <= 0 ? x / y : (x + (y + pow_m1(y >= 0))) / y;}template <typename T, suisen::constraints_t<suisen::is_nbit<T, 16>> = nullptr>__attribute__((target("popcnt"))) constexpr inline int popcount(const T x) { return _mm_popcnt_u32(x); }template <typename T, suisen::constraints_t<suisen::is_nbit<T, 32>> = nullptr>__attribute__((target("popcnt"))) constexpr inline int popcount(const T x) { return _mm_popcnt_u32(x); }template <typename T, suisen::constraints_t<suisen::is_nbit<T, 64>> = nullptr>__attribute__((target("popcnt"))) constexpr inline int popcount(const T x) { return _mm_popcnt_u64(x); }template <typename T, suisen::constraints_t<suisen::is_nbit<T, 16>> = nullptr>constexpr inline int count_lz(const T x) { return x ? __builtin_clz(x) : suisen::bit_num<T>; }template <typename T, suisen::constraints_t<suisen::is_nbit<T, 32>> = nullptr>constexpr inline int count_lz(const T x) { return x ? __builtin_clz(x) : suisen::bit_num<T>; }template <typename T, suisen::constraints_t<suisen::is_nbit<T, 64>> = nullptr>constexpr inline int count_lz(const T x) { return x ? __builtin_clzll(x) : suisen::bit_num<T>; }template <typename T, suisen::constraints_t<suisen::is_nbit<T, 16>> = nullptr>constexpr inline int count_tz(const T x) { return x ? __builtin_ctz(x) : suisen::bit_num<T>; }template <typename T, suisen::constraints_t<suisen::is_nbit<T, 32>> = nullptr>constexpr inline int count_tz(const T x) { return x ? __builtin_ctz(x) : suisen::bit_num<T>; }template <typename T, suisen::constraints_t<suisen::is_nbit<T, 64>> = nullptr>constexpr inline int count_tz(const T x) { return x ? __builtin_ctzll(x) : suisen::bit_num<T>; }template <typename T>constexpr inline int floor_log2(const T x) { return suisen::bit_num<T> -1 - count_lz(x); }template <typename T>constexpr inline int ceil_log2(const T x) { return floor_log2(x) + ((x & -x) != x); }template <typename T>constexpr inline int kth_bit(const T x, const unsigned int k) { return (x >> k) & 1; }template <typename T>constexpr inline int parity(const T x) { return popcount(x) & 1; }// ! containertemplate <typename T, typename Comparator, suisen::constraints_t<suisen::is_comparator<Comparator, T>> = nullptr>auto priqueue_comp(const Comparator comparator) {return std::priority_queue<T, std::vector<T>, Comparator>(comparator);}template <typename Iterable>auto isize(const Iterable& iterable) -> decltype(int(iterable.size())) {return iterable.size();}template <typename T, typename Gen, suisen::constraints_t<suisen::is_same_as_invoke_result<T, Gen, int>> = nullptr>auto generate_vector(int n, Gen generator) {std::vector<T> v(n);for (int i = 0; i < n; ++i) v[i] = generator(i);return v;}template <typename T>auto generate_range_vector(T l, T r) {return generate_vector(r - l, [l](int i) { return l + i; });}template <typename T>auto generate_range_vector(T n) {return generate_range_vector(0, n);}template <typename T>void sort_unique_erase(std::vector<T>& a) {std::sort(a.begin(), a.end());a.erase(std::unique(a.begin(), a.end()), a.end());}template <typename InputIterator, typename BiConsumer>auto foreach_adjacent_values(InputIterator first, InputIterator last, BiConsumer f) -> decltype(f(*first++, *last), void()) {if (first != last) for (auto itr = first, itl = itr++; itr != last; itl = itr++) f(*itl, *itr);}template <typename Container, typename BiConsumer>auto foreach_adjacent_values(Container c, BiConsumer f) -> decltype(c.begin(), c.end(), void()) {foreach_adjacent_values(c.begin(), c.end(), f);}// ! other utilities// x <- min(x, y). returns true iff `x` has chenged.template <typename T>inline bool chmin(T& x, const T& y) {if (y >= x) return false;x = y;return true;}// x <- max(x, y). returns true iff `x` has chenged.template <typename T>inline bool chmax(T& x, const T& y) {if (y <= x) return false;x = y;return true;}template <typename T, std::enable_if_t<std::is_integral_v<T>, std::nullptr_t> = nullptr>std::string bin(T val, int bit_num = -1) {std::string res;if (bit_num >= 0) {for (int bit = bit_num; bit-- > 0;) res += '0' + ((val >> bit) & 1);} else {for (; val; val >>= 1) res += '0' + (val & 1);std::reverse(res.begin(), res.end());}return res;}template <typename T, std::enable_if_t<std::is_integral_v<T>, std::nullptr_t> = nullptr>std::vector<T> digits_low_to_high(T val, T base = 10) {std::vector<T> res;for (; val; val /= base) res.push_back(val % base);if (res.empty()) res.push_back(T{ 0 });return res;}template <typename T, std::enable_if_t<std::is_integral_v<T>, std::nullptr_t> = nullptr>std::vector<T> digits_high_to_low(T val, T base = 10) {auto res = digits_low_to_high(val, base);std::reverse(res.begin(), res.end());return res;}template <typename T>std::string join(const std::vector<T>& v, const std::string& sep, const std::string& end) {std::ostringstream ss;for (auto it = v.begin(); it != v.end();) {ss << *it;if (++it != v.end()) ss << sep;}ss << end;return ss.str();}namespace suisen {}using namespace suisen;using namespace std;struct io_setup {io_setup(int precision = 20) {std::ios::sync_with_stdio(false);std::cin.tie(nullptr);std::cout << std::fixed << std::setprecision(precision);}} io_setup_{};// ! code from here#include <vector>namespace suisen {template <int base_as_int, typename mint>struct static_pow_mods {static_pow_mods() {}static_pow_mods(int n) { ensure(n); }const mint& operator[](int i) const {ensure(i);return pows[i];}static void ensure(int n) {int sz = pows.size();if (sz > n) return;pows.resize(n + 1);for (int i = sz; i <= n; ++i) pows[i] = base * pows[i - 1];}private:static inline std::vector<mint> pows { 1 };static inline mint base = base_as_int;static constexpr int mod = mint::mod();};template <typename mint>struct pow_mods {pow_mods() {}pow_mods(mint base, int n) : base(base) { ensure(n); }const mint& operator[](int i) const {ensure(i);return pows[i];}void ensure(int n) const {int sz = pows.size();if (sz > n) return;pows.resize(n + 1);for (int i = sz; i <= n; ++i) pows[i] = base * pows[i - 1];}private:mutable std::vector<mint> pows { 1 };mint base;static constexpr int mod = mint::mod();};}namespace suisen {namespace internal::rolling_hash {// reference: https://qiita.com/keymoon/items/11fac5627672a6d6a9f6struct Modint2305843009213693951 {using self = Modint2305843009213693951;Modint2305843009213693951() = default;Modint2305843009213693951(uint64_t v) : v(fast_mod(v)) {}static constexpr uint64_t mod() {return _mod;}static constexpr uint64_t fast_mod(uint64_t v) {constexpr uint32_t mid = 61;constexpr uint64_t mask = (uint64_t(1) << mid) - 1;uint64_t u = v >> mid;uint64_t d = v & mask;uint64_t res = u + d;if (res >= _mod) res -= _mod;return res;}uint64_t val() const {return v;}self& operator+=(const self& rhs) {v += rhs.v;if (v >= _mod) v -= _mod;return *this;}self& operator-=(const self& rhs) {if (v < rhs.v) v += _mod;v -= rhs.v;return *this;}self& operator*=(const self& rhs) {static constexpr uint32_t mid31 = 31;static constexpr uint64_t mask31 = (uint64_t(1) << mid31) - 1;uint64_t au = v >> mid31; // < 2^30uint64_t ad = v & mask31; // < 2^31uint64_t bu = rhs.v >> mid31; // < 2^30uint64_t bd = rhs.v & mask31; // < 2^31// a * b// = (au * 2^31 + ad) * (bu * 2^31 + bd)// = au * bu * 2^62 # au * bu * 2^62 ≡ au * bu * 2 < 2^61// + (au * bd + ad * bu) * 2^31 # m := au * bd + ad * bu// # m <= 2 * (2^31 - 1) * (2^30 - 1) = 2^62 - 6 * 2^30 + 2// # m = mu * 2^30 + md (0 <= mu < 2^32, 0 <= md < 2^30)// # m * 2^31 ≡ mu + md * 2^31 < 2^61 + 2^31// + ad * bd # ad * bd <= (2^31 - 1) ** 2 = 2^62 - 2^32 + 1 < 2^62 - 2^31// ≡ au * bu * 2 + mu + md * 2^31 + ad * bd < 2^63static constexpr uint32_t mid30 = 30;static constexpr uint64_t mask30 = (uint64_t(1) << mid30) - 1;uint64_t m = au * bd + ad * bu;uint64_t mu = m >> mid30;uint64_t md = m & mask30;v = fast_mod((au * bu << 1) + mu + (md << 31) + ad * bd);return *this;}friend self operator+(const self& l, const self& r) { return self(l) += r; }friend self operator-(const self& l, const self& r) { return self(l) -= r; }friend self operator*(const self& l, const self& r) { return self(l) *= r; }friend bool operator==(const self& l, const self& r) { return l.v == r.v; }private:static constexpr uint64_t _mod = (uint64_t(1) << 61) - 1; // 2305843009213693951UL : primeuint64_t v;};template <int base_num, typename mint>std::array<mint, base_num> gen_bases() {static std::mt19937_64 rng(std::random_device{}());std::array<mint, base_num> res;for (int i = 0; i < base_num; ++i) {res[i] = rng();while (res[i].val() < 128) res[i] = rng();}return res;}template <int base_num, typename mint>std::array<pow_mods<mint>, base_num> init_pows(const std::array<mint, base_num>& bases) {std::array<pow_mods<mint>, base_num> res;for (int i = 0; i < base_num; ++i) {res[i] = pow_mods<mint>(bases[i], 0);}return res;}}template <int base_num = 1, typename mint = internal::rolling_hash::Modint2305843009213693951>struct RollingHash {private:using default_mint = internal::rolling_hash::Modint2305843009213693951;public:using modint_type = mint;using hash_type = decltype(mint::mod());RollingHash() {}RollingHash(const std::vector<int>& a) : n(a.size()) {for (int base_id = 0; base_id < base_num; ++base_id) {hash[base_id].resize(n + 1);hash[base_id][0] = 0;for (int i = 0; i < n; ++i) hash[base_id][i + 1] = hash[base_id][i] * bases[base_id] + a[i];}}auto operator()(int l, int r) {std::array<mint, base_num> res;for (int base_id = 0; base_id < base_num; ++base_id) {res[base_id] = hash[base_id][r] - hash[base_id][l] * pows[base_id][r - l];}return res;}auto concat(std::array<mint, base_num> h, int l, int r) {for (int base_id = 0; base_id < base_num; ++base_id) {h[base_id] = h[base_id] * pows[base_id][r - l] + hash[base_id][r] - hash[base_id][l] * pows[base_id][r - l];}return h;}auto swapped(int i) {std::array<mint, base_num> res{};res = concat(res, 0, i + 0);res = concat(res, i + 1, i + 2);res = concat(res, i + 0, i + 1);res = concat(res, i + 2, n);return res;}static auto mod() {return mint::mod();}static void set_bases(const std::array<mint, base_num>& new_bases) {bases = new_bases;pows = internal::rolling_hash::init_pows<base_num, mint>(bases);}template <typename Iterable, typename ToIntFunction>static RollingHash from(const Iterable& s, ToIntFunction f) {std::vector<int> a;for (auto&& e : s) a.push_back(f(e));return RollingHash(a);}static RollingHash from_lowercase_alphabet(const std::string& s) {return from(s, [](const auto& e) { return e - 'a' + 1; });}static RollingHash from_uppercase_alphabet(const std::string& s) {return from(s, [](const auto& e) { return e - 'A' + 1; });}static RollingHash from_alphabet(const std::string& s) {return from(s, [](const auto& e) { return std::islower(e) ? e - 'a' + 27 : e - 'A' + 1; });}static RollingHash from_digit(const std::string& s) {return from(s, [](const auto& e) { return e - '0' + 1; });}private:static inline std::array<mint, base_num> bases = internal::rolling_hash::gen_bases<base_num, mint>();static inline std::array<pow_mods<mint>, base_num> pows = internal::rolling_hash::init_pows<base_num, mint>(bases);int n;std::array<std::vector<mint>, base_num> hash;};} // namespace suisenint main() {input(int, n);vector<string> s(n);read(s);using RH = RollingHash<2>;vector<set<array<uint64_t, 2>>> hs(1000001);rep(i, n) {int len = s[i].size();auto h = RH::from_lowercase_alphabet(s[i]);auto hn = h(0, len);if (hs[len].empty()) {print("No");} else {bool found = false;if (hs[len].count({ hn[0].val(), hn[1].val() })) {found = true;} else {rep(j, len - 1) {auto hj = h.swapped(j);if (hs[len].count({ hj[0].val(), hj[1].val() })) {found = true;break;}}}print(found ? "Yes" : "No");}hs[len].insert({ hn[0].val(), hn[1].val() });}return 0;}