結果

問題 No.867 避難経路
ユーザー noshi91noshi91
提出日時 2023-02-28 17:52:38
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 3,475 ms / 6,000 ms
コード長 5,805 bytes
コンパイル時間 2,708 ms
コンパイル使用メモリ 223,104 KB
実行使用メモリ 25,728 KB
最終ジャッジ日時 2024-09-15 20:31:48
合計ジャッジ時間 88,738 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
5,248 KB
testcase_01 AC 4 ms
5,376 KB
testcase_02 AC 4 ms
5,376 KB
testcase_03 AC 4 ms
5,376 KB
testcase_04 AC 4 ms
5,376 KB
testcase_05 AC 4 ms
5,376 KB
testcase_06 AC 4 ms
5,376 KB
testcase_07 AC 4 ms
5,376 KB
testcase_08 AC 4 ms
5,376 KB
testcase_09 AC 4 ms
5,376 KB
testcase_10 AC 4 ms
5,376 KB
testcase_11 AC 4 ms
5,376 KB
testcase_12 AC 3 ms
5,376 KB
testcase_13 AC 4 ms
5,376 KB
testcase_14 AC 4 ms
5,376 KB
testcase_15 AC 3,368 ms
25,728 KB
testcase_16 AC 3,141 ms
25,728 KB
testcase_17 AC 3,475 ms
25,600 KB
testcase_18 AC 3,218 ms
25,728 KB
testcase_19 AC 3,167 ms
25,728 KB
testcase_20 AC 3,152 ms
24,120 KB
testcase_21 AC 3,062 ms
24,380 KB
testcase_22 AC 3,213 ms
22,232 KB
testcase_23 AC 3,443 ms
22,364 KB
testcase_24 AC 3,116 ms
22,440 KB
testcase_25 AC 3,233 ms
20,356 KB
testcase_26 AC 3,257 ms
20,360 KB
testcase_27 AC 3,318 ms
20,492 KB
testcase_28 AC 2,579 ms
20,496 KB
testcase_29 AC 2,636 ms
20,340 KB
testcase_30 AC 2,434 ms
20,184 KB
testcase_31 AC 2,394 ms
22,112 KB
testcase_32 AC 2,425 ms
22,016 KB
testcase_33 AC 2,403 ms
21,948 KB
testcase_34 AC 2,466 ms
20,308 KB
testcase_35 AC 2,547 ms
20,184 KB
testcase_36 AC 2,591 ms
20,180 KB
testcase_37 AC 2,431 ms
20,148 KB
testcase_38 AC 2,397 ms
20,280 KB
testcase_39 AC 2,375 ms
20,276 KB
testcase_40 AC 2,399 ms
20,280 KB
testcase_41 AC 2 ms
5,376 KB
testcase_42 AC 2 ms
5,376 KB
testcase_43 AC 3 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

//#define NDEBUG
#include <cstddef>
#include <cstdint>
#include <iostream>
#include <bits/stdc++.h>

namespace n91 {

using i8 = std::int_fast8_t;
using i32 = std::int_fast32_t;
using i64 = std::int_fast64_t;
using u8 = std::uint_fast8_t;
using u32 = std::uint_fast32_t;
using u64 = std::uint_fast64_t;
using isize = std::ptrdiff_t;
using usize = std::size_t;

constexpr usize operator"" _z(unsigned long long x) noexcept {
  return static_cast<usize>(x);
}

class rep {
  const usize f, l;

public:
  class itr {
    friend rep;
    usize i;
    constexpr itr(const usize x) noexcept : i(x) {}

  public:
    void operator++() noexcept { ++i; }
    constexpr usize operator*() const noexcept { return i; }
    constexpr bool operator!=(const itr x) const noexcept { return i != x.i; }
  };
  constexpr rep(const usize first, const usize last) noexcept
      : f(first), l(last) {}
  constexpr itr begin() const noexcept { return itr(f); }
  constexpr itr end() const noexcept { return itr(l); }
};
class revrep {
  const usize f, l;

public:
  class itr {
    friend revrep;
    usize i;
    constexpr itr(usize x) noexcept : i(x) {}

  public:
    void operator++() noexcept { --i; }
    constexpr usize operator*() const noexcept { return i; }
    constexpr bool operator!=(const itr x) const noexcept { return i != x.i; }
  };
  constexpr revrep(usize first, usize last) noexcept : f(--first), l(--last) {}
  constexpr itr begin() const noexcept { return itr(l); }
  constexpr itr end() const noexcept { return itr(f); }
};
template <class T> using vec_alias = std::vector<T>;
template <class T> auto md_vec(const usize n, const T &value) {
  return std::vector<T>(n, value);
}
template <class... Args> auto md_vec(const usize n, Args... args) {
  return std::vector<decltype(md_vec(args...))>(n, md_vec(args...));
}
template <class T> constexpr T difference(const T &a, const T &b) {
  return a < b ? b - a : a - b;
}
template <class T> T scan() {
  T ret;
  std::cin >> ret;
  return ret;
}

} // namespace n91

#include <algorithm>
#include <array>
#include <iostream>
#include <queue>
#include <utility>

namespace n91 {

void main_() {
  static constexpr u64 MaxK = static_cast<u64>(300);

  const auto h = scan<usize>();
  const auto w = scan<usize>();
  const usize gx = scan<usize>() - 1_z;
  const usize gy = scan<usize>() - 1_z;

  auto a = md_vec(h, w, static_cast<u64>(0));
  for (auto &v : a) {
    for (auto &e : v) {
      e = scan<u64>();
    }
  }

  const usize q = scan<usize>();
  struct query_type {
    usize x, y;
    u64 k;
    u64 ans;
  };
  std::vector<query_type> query(q);
  for (auto &e : query) {
    e.x = scan<usize>() - 1_z;
    e.y = scan<usize>() - 1_z;
    e.k = scan<u64>();
  }
  std::array<std::vector<usize>, static_cast<usize>(MaxK)> small;
  for (const auto i : rep(0_z, q)) {
    if (query[i].k < MaxK) {
      small[query[i].k].push_back(i);
    }
  }

  {
    auto d = md_vec(h, w, std::numeric_limits<u64>::max());
    struct node_type {
      usize x, y;
      u64 c;
      constexpr node_type(const usize x, const usize y, const u64 c)
          : x(x), y(y), c(c) {}
      constexpr bool operator<(const node_type &rhs) const noexcept {
        return rhs.c < c;
      }
    };
    std::priority_queue<node_type> que;
    for (const auto k : rep(1_z, static_cast<usize>(MaxK))) {
      for (auto &v : d) {
        std::fill(v.begin(), v.end(), std::numeric_limits<u64>::max());
      }
      const u64 kk = static_cast<u64>(k * k);
      std::array<usize, 4_z> dx{-1_z, 0_z, 0_z, 1_z}, dy{0_z, -1_z, 1_z, 0_z};
      que.emplace(gx, gy, static_cast<u64>(0));
      d[gx][gy] = static_cast<u64>(0);
      while (!que.empty()) {
        const usize x = que.top().x;
        const usize y = que.top().y;
        const u64 c = que.top().c;
        que.pop();
        if (d[x][y] < c) {
          continue;
        }
        const u64 nc = c + a[x][y] + kk;
        for (const auto i : rep(0_z, 4_z)) {
          const usize nx = x + dx[i];
          const usize ny = y + dy[i];
          if (nx < h && ny < w && d[nx][ny] > nc) {
            d[nx][ny] = nc;
            que.emplace(nx, ny, nc);
          }
        }
      }

      for (const auto i : small[k]) {
        auto &e = query[i];
        e.ans = d[e.x][e.y] + a[e.x][e.y] + kk;
      }
    }
  }

  {
    using p = std::pair<usize, u64>;
    auto d = md_vec(
        h, w,
        p(std::numeric_limits<usize>::max(), std::numeric_limits<u64>::max()));
    std::array<usize, 4_z> dx{-1_z, 0_z, 0_z, 1_z}, dy{0_z, -1_z, 1_z, 0_z};
    struct node_type {
      usize x, y;
      p c;
      constexpr node_type(const usize x, const usize y, const p c)
          : x(x), y(y), c(c) {}
      constexpr bool operator<(const node_type &rhs) const noexcept {
        return rhs.c < c;
      }
    };
    std::priority_queue<node_type> que;
    que.emplace(gx, gy, p(0_z, static_cast<u64>(0)));
    d[gx][gy] = {0_z, static_cast<u64>(0)};
    while (!que.empty()) {
      const usize x = que.top().x;
      const usize y = que.top().y;
      const p c = que.top().c;
      que.pop();
      if (d[x][y] < c) {
        continue;
      }
      const p nc = p(c.first + 1_z, c.second + a[x][y]);
      for (const auto i : rep(0_z, 4_z)) {
        const usize nx = x + dx[i];
        const usize ny = y + dy[i];
        if (nx < h && ny < w && d[nx][ny] > nc) {
          d[nx][ny] = nc;
          que.emplace(nx, ny, nc);
        }
      }
    }
    for (auto &e : query) {
      if (e.k >= MaxK) {
        const usize len = d[e.x][e.y].first + 1_z;
        const u64 dist = d[e.x][e.y].second + a[e.x][e.y];
        e.ans = dist + e.k * e.k * len;
      }
    }
  }

  for (const auto &e : query) {
    std::cout << e.ans << std::endl;
  }
}

} // namespace n91

int main() {
  n91::main_();
  return 0;
}
0