#973879 (C++23) No.2731 Two Colors

提出ソース

結果

問題	No.2731 Two Colors
ユーザー	miscalc
提出日時	2024-04-19 21:41:26
言語	C++23 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	466 ms / 3,000 ms
コード長	12,124 bytes
コンパイル時間	5,864 ms
コンパイル使用メモリ	331,444 KB
実行使用メモリ	20,876 KB
最終ジャッジ日時	2024-10-11 14:29:39
合計ジャッジ時間	13,366 ms
ジャッジサーバーID （参考情報）	judge4 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	461 ms 19,200 KB
testcase_01	AC	462 ms 19,328 KB
testcase_02	AC	466 ms 19,328 KB
testcase_03	AC	5 ms 6,820 KB
testcase_04	AC	23 ms 6,816 KB
testcase_05	AC	15 ms 6,816 KB
testcase_06	AC	62 ms 6,820 KB
testcase_07	AC	74 ms 7,196 KB
testcase_08	AC	12 ms 6,816 KB
testcase_09	AC	277 ms 16,076 KB
testcase_10	AC	5 ms 6,820 KB
testcase_11	AC	257 ms 20,876 KB
testcase_12	AC	275 ms 15,796 KB
testcase_13	AC	216 ms 15,272 KB
testcase_14	AC	119 ms 9,328 KB
testcase_15	AC	11 ms 6,816 KB
testcase_16	AC	103 ms 9,116 KB
testcase_17	AC	153 ms 11,032 KB
testcase_18	AC	26 ms 6,816 KB
testcase_19	AC	107 ms 9,216 KB
testcase_20	AC	183 ms 11,628 KB
testcase_21	AC	28 ms 6,820 KB
testcase_22	AC	280 ms 18,324 KB
testcase_23	AC	67 ms 6,968 KB
testcase_24	AC	33 ms 6,820 KB
testcase_25	AC	3 ms 6,816 KB
testcase_26	AC	223 ms 14,100 KB
testcase_27	AC	280 ms 18,124 KB
testcase_28	AC	173 ms 13,824 KB
testcase_29	AC	53 ms 6,816 KB
testcase_30	AC	100 ms 8,904 KB
testcase_31	AC	61 ms 7,660 KB
testcase_32	AC	335 ms 20,196 KB
testcase_33	AC	2 ms 6,816 KB
testcase_34	AC	2 ms 6,816 KB
testcase_35	AC	2 ms 6,816 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ld = long double;
using ull = unsigned long long;
using pll = pair<ll, ll>;
using tlll = tuple<ll, ll, ll>;
using tllll = tuple<ll, ll, ll, ll>;
using vll = vector<ll>;
using vpll = vector<pll>;
using vtlll = vector<tlll>;
using vtllll = vector<tllll>;
using vstr = vector<string>;
using vvl = vector<vector<ll>>;
constexpr ll INF = 4'000'000'000'000'000'037;
template <class T, class U> bool chmin(T &a, const U &b) { if (a > b) { a = b; return true; } return false; }
template <class T, class U> bool chmax(T &a, const U &b) { if (a < b) { a = b; return true; } return false; }
ll safemod(ll a, ll m) { ll res = a % m; if (res < 0) res += m; return res; }
ll divfloor(ll a, ll b) { if (b < 0) a = -a, b = -b; return (a - safemod(a, b)) / b; }
ll divceil(ll a, ll b) { if (b < 0) a = -a, b = -b; return divfloor(a + b - 1, b); }
ll pow_ll(ll a, ll b) { if (a == 0 || a == 1) return a; if (a == -1) return b & 1 ? -1 : 1; ll res = 1; for (int i = 0; i < b; i++) res *= a; return res; }
ll mul_limited(ll a, ll b, ll m = INF) { return b == 0 ? 0 : a > m / b ? m : a * b; }
ll pow_limited(ll a, ll b, ll m = INF) { if (a == 0 || a == 1) return a; ll res = 1; for (int i = 0; i < b; i++) { if (res > m / a) return m; res *= a; } return res; }
#define overload4(_1, _2, _3, _4, name, ...) name
#define rep1(i, n) for (ll i = 0; i < ll(n); i++)
#define rep2(i, l, r) for (ll i = ll(l); i < ll(r); i++)
#define rep3(i, l, r, d) for (ll i = ll(l); (d) > 0 ? i < ll(r) : i > ll(r); i += d)
#define rep(...) overload4(__VA_ARGS__, rep3, rep2, rep1)(__VA_ARGS__)
#define all(a) (a).begin(), (a).end()
template <class T, class F = decltype(plus<>())> vector<T> cuml(vector<T> v, const F &op = plus<>(), const T &e = 0) { v.emplace_back(e); exclusive_scan(v.begin(), v.end(), v.begin(), e, op); return v; }
template <class T, class F = decltype(plus<>())> vector<T> cumr(vector<T> v, const F &op = plus<>(), const T &e = 0) { v.insert(v.begin(), e); exclusive_scan(v.rbegin(), v.rend(), v.rbegin(), e, op); return v; }
template <class T> vector<T> adjd(vector<T> v) { adjacent_difference(v.begin(), v.end(), v.begin()); return v; }
template <class T = ll> struct max_op { T operator()(const T &a, const T &b) const { return max(a, b); } };
template <class T = ll> struct min_op { T operator()(const T &a, const T &b) const { return min(a, b); } };
struct max_e { ll operator()() const { return -INF; } };
struct min_e { ll operator()() const { return INF; } };
template <class T> vector<T> cumlmax(const vector<T> &v) { return cuml(v, max_op<T>(), max_e()()); }
template <class T> vector<T> cumrmax(const vector<T> &v) { return cumr(v, max_op<T>(), max_e()()); }
template <class T> vector<T> cumlmin(const vector<T> &v) { return cuml(v, min_op<T>(), min_e()()); }
template <class T> vector<T> cumrmin(const vector<T> &v) { return cumr(v, min_op<T>(), min_e()()); }
template <class T> vector<T> sorted(vector<T> v) { sort(v.begin(), v.end()); return v; }
template <class T> vector<T> reversed(const vector<T> &v) { return {v.rbegin(), v.rend()}; }
template <class T> void unique(vector<T> &v) { v.erase(unique(v.begin(), v.end()), v.end()); }
template <class T> void sortunique(vector<T> &v) { sort(v.begin(), v.end()); v.erase(unique(v.begin(), v.end()), v.end()); }
template <class T> void rotate(vector<T> v, int k) { rotate(v.begin(), v.begin() + k, v.end()); }
template <class T> vector<T> rotated(vector<T> v, int k) { rotate(v.begin(), v.begin() + k, v.end()); return v; }
string sorted(string s) { sort(s.begin(), s.end()); return s; }
string reversed(const string &s) { return {s.rbegin(), s.rend()}; }
void unique(string &s) { s.erase(unique(s.begin(), s.end()), s.end()); }
void sortunique(string &s) { sort(s.begin(), s.end()); s.erase(unique(s.begin(), s.end()), s.end()); }
void rotate(string &s, int k) { rotate(s.begin(), s.begin() + k, s.end()); }
string rotated(string s, int k) { rotate(s.begin(), s.begin() + k, s.end()); return s; }
template <class T> vector<vector<T>> transposed(const vector<vector<T>> &a) { const size_t n = a.size(), m = a.at(0).size(); vector<vector<T>> b(m, vector<T>(n)); for (size_t i = 0; i < n; i++) for (size_t j = 0; j < m; j++) b[j][i] = a[i].at(j); return b; }
template <class T, const size_t m> array<vector<T>, m> transposed(const vector<array<T, m>> &a) { const size_t n = a.size(); array<vector<T>, m> b; b.fill(vector<T>(n)); for (size_t i = 0; i < n; i++) for (size_t j = 0; j < m; j++) b[j][i] = a[i][j]; return b; }
template <class T, const size_t n> vector<array<T, n>> transposed(const array<vector<T>, n> &a) { const size_t m = a.at(0).size(); vector<array<T, n>> b(m); for (size_t i = 0; i < n; i++) for (size_t j = 0; j < m; j++) b[j][i] = a[i].at(j); return b; }
template <class T, class U> pair<vector<T>, vector<U>> transposed(const vector<pair<T, U>> &a) { const size_t n = a.size(); vector<T> b(n); vector<U> c(n); for (size_t i = 0; i < n; i++) b[i] = a[i].first, c[i] = a[i].second; return make_pair(b, c); }
template <class T, class U> vector<pair<T, U>> transposed(const pair<vector<T>, vector<U>> &a) { const size_t n = a.first.size(); vector<pair<T, U>> b(n); for (size_t i = 0; i < n; i++) b[i] = make_pair(a.first[i], a.second.at(i)); return b; }
template <class T1, class T2, class T3> tuple<vector<T1>, vector<T2>, vector<T3>> transposed(const vector<tuple<T1, T2, T3>> &a) { const size_t n = a.size(); vector<T1> b(n); vector<T2> c(n); vector<T3> d(n); for (size_t i = 0; i < n; i++) b[i] = get<0>(a[i]), c[i] = get<1>(a[i]), d[i] = get<2>(a[i]); return make_tuple(b, c, d); }
template <class T1, class T2, class T3> vector<tuple<T1, T2, T3>> transposed(const tuple<vector<T1>, vector<T2>, vector<T3>> &a) { const size_t n = get<0>(a).size(); vector<tuple<T1, T2, T3>> b(n); for (size_t i = 0; i < n; i++) b[i] = make_tuple(get<0>(a)[i], get<1>(a).at(i), get<2>(a).at(i)); return b; }
template <class T1, class T2, class T3, class T4> tuple<vector<T1>, vector<T2>, vector<T3>, vector<T4>> transposed(const vector<tuple<T1, T2, T3, T4>> &a) { const size_t n = a.size(); vector<T1> b(n); vector<T2> c(n); vector<T3> d(n); vector<T4> e(n); for (size_t i = 0; i < n; i++) b[i] = get<0>(a[i]), c[i] = get<1>(a[i]), d[i] = get<2>(a[i]), e[i] = get<3>(a[i]); return make_tuple(b, c, d, e); }
template <class T1, class T2, class T3, class T4> vector<tuple<T1, T2, T3, T4>> transposed(const tuple<vector<T1>, vector<T2>, vector<T3>, vector<T4>> &a) { const size_t n = get<0>(a).size(); vector<tuple<T1, T2, T3, T4>> b(n); for (size_t i = 0; i < n; i++) b[i] = make_tuple(get<0>(a)[i], get<1>(a).at(i), get<2>(a).at(i), get<3>(a).at(i)); return b; }
template <class T = ll> vector<T> digitvec(const string &s) { int n = s.size(); vector<T> a(n); for (int i = 0; i < n; i++) a[i] = s[i] - '0'; return a; }
#if __cplusplus < 202002L
ull bit_ceil(ull x) { ull y = 1; while (y < x) y <<= 1; return y; }
ull bit_floor(ull x) { ull y = 1; while (y <= x) y <<= 1; return y >> 1; }
ull bit_width(ull x) { ull y = 1, z = 0; while (y <= x) y <<= 1, z++; return z; }
ull countr_zero(ull x) { return __builtin_ctzll(x); }
ull popcount(ull x) { return __builtin_popcountll(x); }
ull has_single_bit(ull x) { return popcount(x) == 1; }
#endif
ull lsb_index(ull x) { assert(x != 0); return countr_zero(x); }
ull msb_index(ull x) { assert(x != 0); return bit_width(x) - 1; }
ull lsb_mask(ull x) { assert(x != 0); return x & -x; }
ull msb_mask(ull x) { assert(x != 0); return bit_floor(x); }
#if __has_include(<atcoder/all>)
#include <atcoder/all>
using namespace atcoder;
template <class T, internal::is_modint_t<T> * = nullptr> istream &operator>>(istream &is, T &a) { ll v; is >> v; a = v; return is; }
template <class T, internal::is_modint_t<T> * = nullptr> ostream &operator<<(ostream &os, const T &a) { os << a.val(); return os; }
#endif
template <class T, class U> istream &operator>>(istream &is, pair<T, U> &p) { is >> p.first >> p.second; return is; }
template <class T1, class T2, class T3> istream &operator>>(istream &is, tuple<T1, T2, T3> &t) { is >> get<0>(t) >> get<1>(t) >> get<2>(t); return is; }
template <class T1, class T2, class T3, class T4> istream &operator>>(istream &is, tuple<T1, T2, T3, T4> &t) { is >> get<0>(t) >> get<1>(t) >> get<2>(t) >> get<3>(t); return is; }
template <class T, const size_t N> istream &operator>>(istream &is, array<T, N> &a) { for (size_t i = 0; i < N; i++) is >> a[i]; return is; } 
template <class... T> void input(T&... a) { (cin >> ... >> a); }
template <class T> void inputvec(int n, vector<T> &v) { v.resize(n); for (int i = 0; i < n; i++) cin >> v[i]; }
template <class T, class... Ts> void inputvec(int n, vector<T>& v, vector<Ts>&... vs) { inputvec(n, v); inputvec(n, vs...); }
template <class T> void inputvec2(int n, int m, vector<vector<T>> &v) { v.assign(n, vector<T>(m)); for (int i = 0; i < n; i++) for (int j = 0; j < m; j++) cin >> v[i][j]; }
template <class T, class... Ts> void inputvec2(int n, int m, vector<T>& v, vector<Ts>&... vs) { inputvec2(n, m, v); inputvec2(n, m, vs...); }
template <class T, class... Ts> void print(const T& a, const Ts&... b) { cout << a; (cout << ... << (cout << ' ', b)); cout << '\n'; }
#define INT(...) int __VA_ARGS__; input(__VA_ARGS__)
#define LL(...) ll __VA_ARGS__; input(__VA_ARGS__)
#define STR(...) string __VA_ARGS__; input(__VA_ARGS__)
#define VINT(n, ...) vector<int> __VA_ARGS__; inputvec(n, __VA_ARGS__)
#define VLL(n, ...) vector<ll> __VA_ARGS__; inputvec(n, __VA_ARGS__)
#define VSTR(n, ...) vector<string> __VA_ARGS__; inputvec(n, __VA_ARGS__)
#define VPLL(n, ...) vector<pll> __VA_ARGS__; inputvec(n, __VA_ARGS__)
#define VTLLL(n, ...) vector<tlll> __VA_ARGS__; inputvec(n, __VA_ARGS__)
#define VTLLLL(n, ...) vector<tllll> __VA_ARGS__; inputvec(n, __VA_ARGS__)
#define AL(n, ...) array<ll, n> __VA_ARGS__; input(__VA_ARGS__)
#define VAL(n, m, ...) vector<array<ll, m>> __VA_ARGS__; inputvec(n, __VA_ARGS__)
#define VVL(n, m, ...) vector<vector<ll>> __VA_ARGS__; inputvec2(n, m, __VA_ARGS__)
#define FINALANS(x) do { cout << (x) << '\n'; exit(0); } while (false)
template <class T> void printvec(const vector<T> &v) { int n = v.size(); for (int i = 0; i < n; i++) cout << v[i] << (i == n - 1 ? "" : " "); cout << '\n'; }
template <class T> void printvect(const vector<T> &v) { for (auto &vi : v) cout << vi << '\n';}
template <class T> void printvec2(const vector<vector<T>> &v) { for (auto &vi : v) printvec(vi); }
#if __has_include(<atcoder/all>)
using mint = modint998244353;
//using mint = modint1000000007;
//using mint = modint;
#define MINT(...) mint __VA_ARGS__; input(__VA_ARGS__);
#define VMINT(n, ...) vector<mint> __VA_ARGS__; inputvec(n, __VA_ARGS__)
#endif
#ifdef LOCAL // https://zenn.dev/sassan/articles/19db660e4da0a4
#include "/Users/Shared/cpp-dump-main/dump.hpp"
#define dump(...) cpp_dump(__VA_ARGS__)
CPP_DUMP_DEFINE_EXPORT_OBJECT(mint, val());
CPP_DUMP_DEFINE_DANGEROUS_EXPORT_OBJECT(val());
#else
#define dump(...)
#endif

const vpll dij = {{1, 0}, {0, 1}, {-1, 0}, {0, -1}};

int main()
{
  LL(H, W);
  VVL(H, W, A);
  vvl C(H, vll(W, -1));
  auto ok = [&](ll i, ll j) -> bool
  { return 0 <= i && i < H && 0 <= j && j < W; };
  array<priority_queue<tlll, vector<tlll>, greater<tlll>>, 2> pque;
  array<pll, 2> sij = {{{0, 0}, {H - 1, W - 1}}};
  rep(k, 2)
  {
    auto [si, sj] = sij[k];
    C.at(si).at(sj) = k;
    for (auto [di, dj] : dij)
    {
      ll ni = si + di, nj = sj + dj;
      if (!ok(ni, nj))
        continue;
      pque[k].push({A.at(ni).at(nj), ni, nj});
    }
  }
  for (ll k = 0;; k++)
  {
    dump(C);
    // dump(k, A, C, pque);
    ll c = k % 2;
    while (!pque[c].empty())
    {
      auto [a, i, j] = pque[c].top();
      pque[c].pop();
      if (C.at(i).at(j) == c)
        continue;
      C.at(i).at(j) = c;
      for (auto [di, dj] : dij)
      {
        ll ni = i + di, nj = j + dj;
        if (!ok(ni, nj))
          continue;
        if (C.at(ni).at(nj) == c)
          continue;
        if (C.at(ni).at(nj) == (c ^ 1))
          FINALANS(k + 1);
        pque[c].push({A.at(ni).at(nj), ni, nj});
      }
      break;
    }
  }
}

yukicoder

結果

テストケース

ソースコード