結果

問題 No.3082 Make Palindromic Multiple(Judge)
コンテスト
ユーザー 👑 binap
提出日時 2025-02-26 13:49:04
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 6,160 bytes
コンパイル時間 4,239 ms
コンパイル使用メモリ 262,928 KB
実行使用メモリ 11,012 KB
最終ジャッジ日時 2025-03-28 01:21:33
合計ジャッジ時間 11,146 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 63 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#include<atcoder/all>
#define rep(i,n) for(int i=0;i<n;i++)
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef vector<int> vi;
typedef vector<long long> vl;
typedef vector<vector<int>> vvi;
typedef vector<vector<long long>> vvl;
typedef long double ld;
typedef pair<int, int> P;

template <int m> ostream& operator<<(ostream& os, const static_modint<m>& a) {os << a.val(); return os;}
template <int m> ostream& operator<<(ostream& os, const dynamic_modint<m>& a) {os << a.val(); return os;}
template <int m> istream& operator>>(istream& is, static_modint<m>& a) {long long x; is >> x; a = x; return is;}
template <int m> istream& operator>>(istream& is, dynamic_modint<m>& a) {long long x; is >> x; a = x; return is;}
template<typename T> istream& operator>>(istream& is, vector<T>& v){int n = v.size(); assert(n > 0); rep(i, n) is >> v[i]; return is;}
template<typename U, typename T> ostream& operator<<(ostream& os, const pair<U, T>& p){os << p.first << ' ' << p.second; return os;}
template<typename T> ostream& operator<<(ostream& os, const vector<T>& v){int n = v.size(); rep(i, n) os << v[i] << (i == n - 1 ? "\n" : " "); return os;}
template<typename T> ostream& operator<<(ostream& os, const vector<vector<T>>& v){int n = v.size(); rep(i, n) os << v[i] << (i == n - 1 ? "\n" : ""); return os;}
template<typename T> ostream& operator<<(ostream& os, const set<T>& se){for(T x : se) os << x << " "; os << "\n"; return os;}
template<typename T> ostream& operator<<(ostream& os, const unordered_set<T>& se){for(T x : se) os << x << " "; os << "\n"; return os;}
template<typename S, auto op, auto e> ostream& operator<<(ostream& os, const atcoder::segtree<S, op, e>& seg){int n = seg.max_right(0, [](S){return true;}); rep(i, n) os << seg.get(i) << (i == n - 1 ? "\n" : " "); return os;}
template<typename S, auto op, auto e, typename F, auto mapping, auto composition, auto id> ostream& operator<<(ostream& os, const atcoder::lazy_segtree<S, op, e, F, mapping, composition, id>& seg){int n = seg.max_right(0, [](S){return true;}); rep(i, n) os << seg.get(i) << (i == n - 1 ? "\n" : " "); return os;}

template<typename T> void chmin(T& a, T b){a = min(a, b);}
template<typename T> void chmax(T& a, T b){a = max(a, b);}

random_device rd;
mt19937 mt(rd());

int random_prime(){
    while (true){
        int pr = mt() % 100000000 + 900000000;
        if (atcoder::internal::is_prime_constexpr(pr)){
            return pr;
        }
    }
}

using ull = unsigned long long;

using mint1 = atcoder::dynamic_modint<1>;
using mint2 = atcoder::dynamic_modint<2>;

struct mints{
	mint1 d1;
	mint2 d2;
	mints(long long val = 0) : d1(val), d2(val) {}
	mints(mint1 d1, mint2 d2) : d1(d1), d2(d2) {}
	mints& operator++() {
		d1++; d2++;
		return *this;
	}
	mints& operator--() {
		d1--; d2--;
		return *this;
	}
	mints& operator+=(const mints& a) {
		d1 += a.d1;	d2 += a.d2;
		return *this;
	}
	mints& operator-=(const mints& a) {
		d1 -= a.d1;	d2 -= a.d2;
		return *this;
	}
	mints& operator*=(const mints& a) {
		d1 *= a.d1;	d2 *= a.d2;
		return *this;
	}
	mints& operator/=(const mints& a) { return *this = *this * a.inv(); }
	mints operator+() const { return *this; }
	mints operator-() const { return mints(0) - *this; }
	
	mints pow(long long n) const {
		return mints(d1.pow(n), d2.pow(n));
	}
	mints inv() const {
		return mints(d1.inv(), d2.inv());
	}
	
	mints operator+(const mints& a) const {
		return mints(d1 + a.d1, d2 + a.d2);
	}
	mints operator-(const mints& a) const {
		return mints(d1 - a.d1, d2 - a.d2);
	}
	mints operator*(const mints& a) const {
		return mints(d1 * a.d1, d2 * a.d2);
	}
	mints operator/(const mints& a) const {
		return mints(d1 / a.d1, d2 / a.d2);
	}
	bool operator==(const mints& a) const {
		return d1 == a.d1 and d2 == a.d2;
	}
	bool operator!=(const mints& a) const {
		return d1 != a.d1 or d2 != a.d2;
	}
    ull val(){
        return (ull(d1.val()) << 32) + ull(d2.val());
    };
};
ostream& operator<<(ostream& os, const mints& a) {os << a.d1 << '&' << a.d2; return os;}

void random_mod(){
	const int p1 = 998244353;
	const int p2 = 1000000007;
	mint1::set_mod(p1);
	mint2::set_mod(p2);
}

template<typename T>
pair<T, T> pow_mod(T A, T N, T MOD){
	assert(N >= 0);
	if(N == 0) return {1 % MOD, 0};
	if(N == 1) return {A % MOD, 1 % MOD};
	A %= MOD;
	if(N % 2 == 0){
		auto [p, s] = pow_mod(A, N / 2, MOD);
		T p2 = (p * p) % MOD;
		T s2 = (s + s * p) % MOD;
		return {p2, s2};
	}else{
		auto [p, s] = pow_mod(A, N - 1, MOD);
		T p2 = (p * A) % MOD;
		T s2 = (s + p) % MOD;
		return {p2, s2};
	}
}

long long MOD;

int K;
vector<string> S;
vector<long long> T;

void input(){
	cin >> K;
	S.resize(K);
	T.resize(K);
	rep(i, K) cin >> S[i] >> T[i];
}

int primitive(){
	int g = 0;
	
	auto factor = [&](int x){
		assert(x >= 1);
		vector<int> res;
		for(int y = 1; y * y <= x; y++){
			if(x % y == 0){
				res.push_back(y);
				res.push_back(x / y);
			}
		}
		return res;
	};
	
	int p1 = mint1::mod();
	int p2 = mint2::mod();
	vector<int> factor1 = factor(p1- 1);
	vector<int> factor2 = factor(p2 - 1);
	
	while(true){
		g = mt() % min(p1 - 1, p2 - 1) + 1;
		bool success = true;
		for(int y : factor1){
			if(y == p1 - 1) continue;
			auto h = pow_mod<long long>(g, y, p1).first;
			if(h == 1) success = false;
		}
		for(int y : factor2){
			if(y == p2 - 1) continue;
			auto h = pow_mod<long long>(g, y, p2).first;
			if(h == 1) success = false;
		}
		if(success) break;
	}
	return g;
}

bool judge_palindrome(){
	
	
	using F = mints;
	vector<F> hash(2, 0);
	F base(primitive());
	
	auto get_hash = [&]() -> F{
		F x(0);
		rep(i, K){
			F p = F(base).pow(int(S[i].size()));
			F q = p.pow(T[i]);
			x *= q;
			
			F y(0);
			for(auto c : S[i]){
				y *= base;
				y += c - '0' + 1;
			}
			x += y * (q - 1) / (p - 1);
		}
		return x;
	};
	
	rep(t, 2){
		hash[t] = get_hash();
		
		reverse(S.begin(), S.end());
		reverse(T.begin(), T.end());
		rep(i, K) reverse(S[i].begin(), S[i].end());
	}
	return hash[0] == hash[1];
}

int main() {
	random_mod();
	input();
	bool is_palindrome = judge_palindrome();
	if(is_palindrome) cout << "Yes\n";
	else cout << "No\n";
	return 0;
}
0