結果
問題 | No.3121 Prime Dance |
ユーザー |
![]() |
提出日時 | 2025-04-19 02:42:34 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 44 ms / 2,000 ms |
コード長 | 3,709 bytes |
コンパイル時間 | 4,019 ms |
コンパイル使用メモリ | 311,840 KB |
実行使用メモリ | 20,188 KB |
最終ジャッジ日時 | 2025-04-19 02:42:40 |
合計ジャッジ時間 | 5,271 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 21 |
ソースコード
#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 ^= 0x2 ^ 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 ^= 0x2 ^ 4) { if (lpf[x] == 0) { lpf[x] = x; primes.push_back(x); } } return {move(primes), move(lpf)}; } constexpr int PSIZE = 1000000; auto [primes, lpf] = primes_lpf(PSIZE); constexpr int INF = 1001001001; char s[40][40]; int dist[40][40][1600]; bool done[40][40][1600]; } int main() { ios::sync_with_stdio(false); cin.tie(0); int h, w; cin >> h >> w; int xs, ys, xt, yt; cin >> xs >> ys >> xt >> yt; xs--, ys--, xt--, yt--; bool flip = ys == yt; if (flip) swap(xs, ys), swap(xt, yt); rep(i, h) rep(j, w) { char c; cin >> c; if (c == 'S' || c == 'G') c = '.'; if (flip) s[j][i] = c; else s[i][j] = c; } if (flip) swap(h, w); rep(i, 40) rep(j, 40) rep(k, 1600) dist[i][j][k] = INF; deque<tuple<int, int, int>> q; dist[xs][ys][0] = 0; q.emplace_back(xs, ys, 0); while (q.size()) { auto [i, j, k] = q.front(); q.pop_front(); if (done[i][j][k]) continue; done[i][j][k] = true; int d = dist[i][j][k]; if (i + 1 < h && s[i+1][j] == '.' && k + 1 < 1600 && chmin(dist[i+1][j][k+1], d)) q.emplace_front(i+1, j, k+1); if (i && s[i-1][j] == '.' && k + 1 < 1600 && chmin(dist[i-1][j][k+1], d)) q.emplace_front(i-1, j, k+1); if (j + 1 < w && s[i][j+1] == '.' && chmin(dist[i][j+1][k], d+1)) q.emplace_back(i, j+1, k); if (j && s[i][j-1] == '.' && chmin(dist[i][j-1][k], d+1)) q.emplace_back(i, j-1, k); } auto is_prime = [](int x) { return x >= 2 && lpf[x] == x; }; int ans = INF; rep(k, 1600) if (k) { int d = dist[xt][yt][k]; if (d == INF) continue; int dx = abs(xt - xs), dy = abs(yt - ys); assert(dx % 2 == k % 2 && dy % 2 == d % 2); int xsmall = (k - dx) / 2, ysmall = (d - dy) / 2; int cost = 0; for (auto [x, d] : {pair(xsmall, dx), pair(ysmall, dy)}) { if (d % 2) { if (x > 2 || !is_prime(2 + d)) { cost = INF; break; } cost += 2 + 2 + d; } else { for (int p : primes | views::drop(lower_bound(all(primes), x) - primes.begin())) { if (is_prime(p + d)) { cost += p + p + d; break; } } } } chmin(ans, cost); } if (ans == INF) ans = -1; cout << ans << '\n'; }