結果
問題 | No.2207 pCr検査 |
ユーザー | Kude |
提出日時 | 2023-02-03 22:31:24 |
言語 | C++17(gcc12) (gcc 12.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2,710 ms / 3,000 ms |
コード長 | 4,055 bytes |
コンパイル時間 | 2,733 ms |
コンパイル使用メモリ | 233,740 KB |
実行使用メモリ | 124,084 KB |
最終ジャッジ日時 | 2024-07-02 20:38:07 |
合計ジャッジ時間 | 46,403 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
#include<bits/stdc++.h> namespace { #pragma GCC diagnostic ignored "-Wunused-function" #include<atcoder/all> #pragma GCC diagnostic warning "-Wunused-function" using namespace std; using namespace atcoder; #define rep(i,n) for(int i = 0; i < (int)(n); i++) #define rrep(i,n) for(int i = (int)(n) - 1; i >= 0; i--) #define all(x) begin(x), end(x) #define rall(x) rbegin(x), rend(x) template<class T> bool chmax(T& a, const T& b) { if (a < b) { a = b; return true; } else return false; } template<class T> bool chmin(T& a, const T& b) { if (b < a) { a = b; return true; } else return false; } using ll = long long; using P = pair<int,int>; using VI = vector<int>; using VVI = vector<VI>; using VL = vector<ll>; using VVL = vector<VL>; pair<vector<int>, vector<int>> primes_lpf(const int n) { vector<int> primes; primes.reserve(n / 10); vector<int> lpf(n + 1); for (int i = 2; i <= n; i += 2) lpf[i] = 2; for (int i = 3; i <= n; i += 6) lpf[i] = 3; if (2 <= n) primes.push_back(2); if (3 <= n) primes.push_back(3); // 5 * x <= n, x <= floor(n / 5) const int n5 = n / 5; int x = 5; char add_next = 2; for (; x <= n5; x += add_next, add_next ^= 2 ^ 4) { int px = lpf[x]; if (px == 0) { lpf[x] = px = x; primes.push_back(x); } for (int i = 2;; ++i) { int q = primes[i]; int y = q * x; if (y > n) break; lpf[y] = q; if (q == px) break; } } for (; x <= n; x += add_next, add_next ^= 2 ^ 4) { if (lpf[x] == 0) { lpf[x] = x; primes.push_back(x); } } return {move(primes), move(lpf)}; } constexpr int PSIZE = 10000010; auto [primes, lpf] = primes_lpf(PSIZE); vector<pair<int, int>>& factorize(int x) { static vector<pair<int, int>> fs; fs.clear(); while (x != 1) { int p = lpf[x], c = 0; do {x /= p; c++;} while (x % p == 0); fs.emplace_back(p, c); } return fs; } class mint61 { using ull = unsigned long long; using ui128 = __uint128_t; public: static constexpr unsigned long long mod = (1ULL << 61) - 1; ull v = 0; explicit constexpr mint61() {} explicit constexpr mint61(ull x) { v = (x >> 61) + (x & mod); if (v >= mod) v -= mod; } static constexpr mint61 raw(ull x) { mint61 res; res.v = x; return res; } friend constexpr mint61 operator+(mint61 lhs, mint61 rhs) { auto res = lhs.v + rhs.v; return raw(res < mod ? res : res - mod); } friend constexpr mint61 operator-(mint61 lhs, mint61 rhs) { return raw(lhs.v >= rhs.v ? lhs.v - rhs.v : mod + lhs.v - rhs.v); } static constexpr ull multiply_loose_mod(mint61 lhs, mint61 rhs) { // ab = q(m+1)+r = qm+q+r // q+r = ab-qm = ab-floor(ab/(m+1))m < ab-(ab/(m+1)-1)m = ab/(m+1)+m <= (m-1)^2/(m+1)+m = 2m-1-2/(m+1) auto mul = ui128(lhs.v) * rhs.v; return ull(mul >> 61) + ull(mul & mod); } friend constexpr mint61 operator*(mint61 lhs, mint61 rhs) { auto res = multiply_loose_mod(lhs, rhs); return raw(res < mod ? res : res - mod); } mint61& operator+=(mint61 rhs) { return *this = *this + rhs; } mint61& operator-=(mint61 rhs) { return *this = *this - rhs; } mint61& operator*=(mint61 rhs) { return *this = *this * rhs; } friend ostream& operator<<(ostream& os, mint61 x) { return os << x.v; } }; mint61 hsh[10000010]; } int main() { ios::sync_with_stdio(false); static auto RNG = std::mt19937(std::chrono::system_clock::now().time_since_epoch().count()); static auto Distribution = std::uniform_int_distribution<unsigned long long>(2, mint61::mod - 1); cin.tie(0); rep(i, 10000010) hsh[i] = mint61::raw(Distribution(RNG)); mint61 now, tgt; int k; cin >> k; int p; rep(i, k) { int e; cin >> p >> e; tgt += hsh[p] * mint61::raw(e); } for(int r = 1; r < p; r++) { for(auto [p, c]: factorize(p + 1 - r)) { rrep(_, c) now += hsh[p]; } for(auto [p, c]: factorize(r)) { rrep(_, c) now -= hsh[p]; } if (now.v == tgt.v) { cout << p << ' ' << r << '\n'; return 0; } } cout << -1 << ' ' << -1 << '\n'; }