結果
| 問題 |
No.3121 Prime Dance
|
| コンテスト | |
| ユーザー |
 hardenisthegoat
|
| 提出日時 | 2025-04-18 22:38:12 |
| 言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
WA
|
| 実行時間 | - |
| コード長 | 6,749 bytes |
| コンパイル時間 | 3,457 ms |
| コンパイル使用メモリ | 239,392 KB |
| 実行使用メモリ | 12,148 KB |
| 最終ジャッジ日時 | 2025-04-18 22:38:17 |
| 合計ジャッジ時間 | 3,270 ms |
|
ジャッジサーバーID （参考情報） |
judge1 / judge5 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 |
| other | AC * 3 WA * 18 |
ソースコード
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
#define ar array
#define int long long
#define ll long long
#define ld long double
#define pb push_back
#define eb emplace_back
#define pob pop_back()
#define nn "\n"
#define onn << "\n"
#define ee endl
#define oee << endl
#define ss " "
#define oss << " "
#define sei unordered_set<int, custom_hash>
#define ses unordered_set<string>
#define ma unordered_map
#define maii unordered_map<int, int, custom_hash>
#define vc vector
#define id1 int n; cin >> n;
#define gls getline(cin, s)
#define sinf 1e9
#define inf (int) 1e18
#define co cout <<
#define sp(i) setprecision(i) <<
#define f1(i, a) for (auto& i : a)
#define f2(i, j, a) for (auto& [i, j] : a)
#define f3(i, j, k, a) for (auto& [i, j, k] : a)
#define be(s) s.begin(), s.end()
#define rbe(s) s.rbegin(), s.rend()
#define bb(s) s.begin(), s.begin()
#define ctz(x) __builtin_ctzll(x)
#define cb(x) __builtin_popcountll(x)
#define ff(i, begin, end) for (__typeof(end) i = (begin) - ((begin) > (end)); i != (end) - ((begin) > (end)); i += 1 - 2 * ((begin) > (end)))
#define i128 __int128_t
#define lb lower_bound
#define ub upper_bound
#define ook order_of_key
#define fbo find_by_order
#define fn function
#define bs bitset
#define rep(i, a, b) for(int i = a; i < (b); ++i)
#define all(x) begin(x), end(x)
#define sz(x) (int)(x).size()
typedef unsigned long long ull;
typedef pair<int, string> pis;
typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<bool> vb;
typedef vector<string> vs;
typedef vector<double> vd;
typedef vector<pair<int,int>> vp;
typedef vector<vector<int>> vvi;
#pragma GCC optimize("Ofast,unroll-loops")
#ifdef LOCAL
#include "debug_util.h"
#else
#define help(...)
#define helpArray(...)
#define io
#define _time
#define _siu
#endif
template<class T> using pq = priority_queue<T>;
template<class T> using pqg = priority_queue<T, vector<T>, greater<T>>;
template< typename T1, typename T2 >
ostream &operator<<(ostream &os, const pair< T1, T2 >& p) { os << p.first << " " << p.second; return os;}
template< typename T1, typename T2 >
istream &operator>>(istream &is, pair< T1, T2 > &p) {is >> p.first >> p.second; return is;}
template< typename T >
ostream &operator<<(ostream &os, const vector< T > &v) { for(int i = 0; i < (int) v.size(); i++) {os << v[i] << (i + 1 != v.size() ? " " : "");} return os;}
template< typename T >
istream &operator>>(istream &is, vector< T > &v) { for(T &in : v) is >> in; return is;}
struct custom_hash {
static uint64_t splitmix64(uint64_t x) {
// http://xorshift.di.unimi.it/splitmix64.c
x += 0x9e3779b97f4a7c15;
x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9;
x = (x ^ (x >> 27)) * 0x94d049bb133111eb;
return x ^ (x >> 31);
}
size_t operator()(uint64_t x) const {
static const uint64_t FIXED_RANDOM = chrono::steady_clock::now().time_since_epoch().count();
return splitmix64(x + FIXED_RANDOM);
}
};
template<class T>
bool constexpr chmin(T &a, T b) {
if (a > b) {a = b;return true;}
return false;
}
template<class T>
bool constexpr chmax(T &a, T b) {
if (a < b) {a = b;return true;}
return false;
}
// stable sort
template <typename T>
vector<int> argsort(const vector<T> &A) {
vector<int> ans(sz(A));
iota(be(ans), 0);
sort(be(ans),
[&](int i, int j) { return (A[i] == A[j] ? i < j : A[i] < A[j]); });
return ans;
}
//64b
long long lsb(long long x) { return x & -x;}
long long msb(long long x) { return (1ll << (63-__builtin_clzll(x))) & x;}
void onb(long long& x, int i) { x |= (1ll << i);}
void offb(long long& x, int i) { x &= ~(1ll << i);}
bool hb(long long x, int i) { return ((1ll << i) & x) != 0;}
long long bit(long long x) { return 1ll << x;}
void YES(bool t = 1) { cout << (t ? "YES" : "NO") << "\n"; }
void NO(bool t = 1) { YES(!t); }
void Yes(bool t = 1) { cout << (t ? "Yes" : "No") << "\n"; }
void No(bool t = 1) { Yes(!t); }
template<typename T>
vector<pair<T, int>> pfy(vector<T>& f) {
vector<pair<T, int>> v;
for (int i = 0; i < f.size(); i++) {
v.push_back({f[i], i});
}
return v;
}
bool inrange(int i, int j, int limi, int limj) {
return i >= 0 && i < limi && j >= 0 && j < limj;
}
typedef tuple<int,int,int> iii;
vector<char> cao = {'C','D','H','S'};
int dr8[] = {0,-1,0, 1, 1, 1, -1, -1};
int dc8[] = {-1, 0, 1, 0, 1, -1, 1, -1};
//n,w,e,s
int dr[] = {-1, 0, 0, 1};
int dc[] = {0,-1,1, 0};
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
int rand(int l, int r)
{
uniform_int_distribution<int> uid(l, r);
return uid(rng);
}
// Define a tree-based set with integer keys
template <typename T>
using Tee = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
vector<int> spf;
vector<int> primes;
void sieve(int N) {
spf.assign(N, 0);
for(int i = 2; i < N; i++) {
if (spf[i] == 0) spf[i] = i, primes.push_back(i);
int siz = primes.size();
for (int j = 0; j < siz && i * primes[j] < N && primes[j] <= spf[i]; j++) {
spf[i * primes[j]] = primes[j];
}
}
}
void sui() {
auto isprime = [&](int cur) {
return spf[cur] == cur;
};
int h,w;
cin>>h>>w;
vi a(4);
cin>>a;
f1(i,a) --i;
vs v(h);
cin>>v;
vc<vb> vis(h, vb(w));
ff(i,0,h) ff(j,0,w) if (v[i][j] == '#') vis[i][j] = 1;
vis[a[0]][a[1]] = 1;
queue<pii> q;
q.push({a[0], a[1]});
bool b = 0;
while (!q.empty()) {
auto [c,d] = q.front();
q.pop();
if (c == a[2] && d == a[3]) {
b = 1;
break;
}
ff(i,0,4) {
if (inrange(c+dr[i], d+dc[i],h,w)) {
if (!vis[c+dr[i]][d+dc[i]]) {
vis[c+dr[i]][d+dc[i]] = 1;
q.push({c+dr[i], d+dc[i]});
}
}
}
}
help(b);
if (!b) {
co -1 onn;
} else {
int can = 0;
int tot = 0;
int p1 = abs(a[2]-a[0]);
ff(i,0,primes.size()-1) {
// if (primes[i]+p1 < primes[i+1]) {
// co -1 onn;
// return;
// }
if (isprime(primes[i]+p1)) {
tot += 2*primes[i]+p1;
++can;
break;
}
}
int p2 = abs(a[3]-a[1]);
ff(i,0,primes.size()-1) {
// if (primes[i]+p2 < primes[i+1]) {
// co -1 onn;
// return;
// }
if (isprime(primes[i]+p2)) {
tot += 2*primes[i]+p2;
++can;
break;
}
}
if (can != 2) {
co -1 onn;
return;
}
co tot onn;
}
}
signed main() {
cin.tie(NULL);
ios_base::sync_with_stdio(false);
auto start = std::chrono::high_resolution_clock::now();
io;
int n = 1;
// cin >> n;
sieve(1e6+5);
ff(i,0,n)
sui();
_time;
return 0;
}