結果

問題 No.1638 Robot Maze
コンテスト
ユーザー first_vil
提出日時 2021-08-08 13:36:13
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 4,394 bytes
コンパイル時間 2,498 ms
コンパイル使用メモリ 206,628 KB
最終ジャッジ日時 2025-01-23 16:58:39
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ld = long double;
template<class T> using V = vector<T>;
using VI = V<int>;
using VL = V<ll>;
using VS = V<string>;
template<class T> using PQ = priority_queue<T, V<T>, greater<T>>;
using graph = V<VI>;
template<class T> using w_graph = V<V<pair<int, T>>>;
#define FOR(i,a,n) for(int i=(a);i<(n);++i)
#define eFOR(i,a,n) for(int i=(a);i<=(n);++i)
#define rFOR(i,a,n) for(int i=(n)-1;i>=(a);--i)
#define erFOR(i,a,n) for(int i=(n);i>=(a);--i)
#define all(a) a.begin(),a.end()
#define rall(a) a.rbegin(),a.rend()
#define out(y,x) ((y)<0||h<=(y)||(x)<0||w<=(x))
#ifdef _DEBUG
#define line cout << "-----------------------------\n"
#define stop system("pause")
#endif
constexpr ll INF = 1000000000;
constexpr ll LLINF = 1LL << 61;
constexpr ll mod = 1000000007;
constexpr ll MOD = 998244353;
constexpr ld eps = 1e-10;
template<class T>inline bool chmax(T& a, const T& b) { if (a < b) { a = b; return true; }return false; }
template<class T>inline bool chmin(T& a, const T& b) { if (a > b) { a = b; return true; }return false; }
inline void init() { cin.tie(nullptr); cout.tie(nullptr); ios::sync_with_stdio(false); cout << fixed << setprecision(15); }
template<class T>inline istream& operator>>(istream& is, V<T>& v) { for (auto& a : v)is >> a; return is; }
template<class T, class U>inline istream& operator>>(istream& is, pair<T, U>& p) { is >> p.first >> p.second; return is; }
template<class T>inline V<T> vec(size_t a) { return V<T>(a); }
template<class T>inline V<T> defvec(T def, size_t a) { return V<T>(a, def); }
template<class T, class... Ts>inline auto vec(size_t a, Ts... ts) { return V<decltype(vec<T>(ts...))>(a, vec<T>(ts...)); }
template<class T, class... Ts>inline auto defvec(T def, size_t a, Ts... ts) { return V<decltype(defvec<T>(def, ts...))>(a, defvec<T>(def, ts...)); }
template<class T>inline void print(const T& a) { cout << a << "\n"; }
template<class T, class... Ts>inline void print(const T& a, const Ts&... ts) { cout << a << " "; print(ts...); }
template<class T>inline void print(const V<T>& v) { for (int i = 0; i < v.size(); ++i)cout << v[i] << (i == v.size() - 1 ? "\n" : " "); }
template<class T>inline void print(const V<V<T>>& v) { for (auto& a : v)print(a); }
template<class T>inline T sum(const V<T>& a, int l, int r) { return 0 <= l && l <= r && r < a.size() ? a[r] - (l == 0 ? 0 : a[l - 1]) : 0; }
template<class T>inline void END(T s) { print(s); exit(0); }
void END() { exit(0); }

template<int M> class Indexer {
    static_assert(M > 0, "M must be positive");
    array<int, M> lim;
    int siz;
public:
    Indexer(initializer_list<int> lim_init)  {
        assert(lim_init.size() == M);
        copy(all(lim_init), lim.begin());
        siz = 1;
        FOR(i, 0, M) {
            assert(lim[i] > 0);
            siz *= lim[i];
        }
    };
    int operator() (initializer_list<int> target_init) {
        assert(target_init.size() == M);
        array<int, M> target;
        copy(all(target_init), target.begin());
        int res = 0;
        FOR(i, 0, M)res = res * lim[i] + target[i];
        return res;
    }
    array<int, M> operator() (int a) {
        assert(a < size());
        array<int, M> res;
        rFOR(i, 0, M) {
            res[i] = a % lim[i];
            a /= lim[i];
        }
        return res;
    }
    int size() {
        return siz;
    }
};

int main() {
    init();

    int h, w; cin >> h >> w;
    VI dir(4); cin >> dir;
    ll k, p; cin >> k >> p;
    int sy, sx, gy, gx; cin >> sy >> sx >> gy >> gx;
    --sy, --sx, --gy, --gx;
    VS c(h); cin >> c;

    Indexer<2> indexer({ h,w });
    VL dp(indexer.size(), LLINF);
    int start = indexer({ sy,sx });
    dp[start] = 0;
    PQ<pair<ll, int>> dik;
    dik.emplace(0, start);
    int dy[4] = { -1,1,0,0 }, dx[4] = { 0,0,1,-1 };
    while (!dik.empty()) {
        auto [d, cur] = dik.top();
        dik.pop();
        if (dp[cur] < d)continue;
        auto [y, x] = indexer(cur);
        FOR(i, 0, 4) {
            int ny = y + dy[i], nx = x + dx[i];
            if (out(ny, nx) || c[ny][nx] == '#')continue;
            ll cost = dir[i] + p * (c[ny][nx] == '@');
            int to = indexer({ ny,nx });
            if (!chmin(dp[to], d + cost))continue;
            dik.emplace(d + cost, to);
        }
    }
    print(dp[indexer({ gy,gx })] <= k ? "Yes" : "No");

    return 0;
}
0