結果

問題 No.34 砂漠の行商人
ユーザー kjnh10kjnh10
提出日時 2020-06-04 21:46:21
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 5,090 bytes
コンパイル時間 2,735 ms
コンパイル使用メモリ 217,080 KB
実行使用メモリ 166,708 KB
最終ジャッジ日時 2024-05-07 14:00:42
合計ジャッジ時間 14,903 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 9 ms
10,112 KB
testcase_03 AC 12 ms
8,960 KB
testcase_04 AC 75 ms
42,368 KB
testcase_05 AC 85 ms
55,936 KB
testcase_06 AC 18 ms
33,480 KB
testcase_07 AC 130 ms
82,048 KB
testcase_08 AC 181 ms
111,104 KB
testcase_09 AC 3,218 ms
94,592 KB
testcase_10 TLE -
testcase_11 AC 3,313 ms
161,152 KB
testcase_12 AC 78 ms
43,904 KB
testcase_13 TLE -
testcase_14 TLE -
testcase_15 AC 1,643 ms
49,536 KB
testcase_16 AC 1,544 ms
46,080 KB
testcase_17 AC 1,911 ms
92,160 KB
testcase_18 AC 311 ms
12,416 KB
testcase_19 AC 1,238 ms
145,792 KB
testcase_20 AC 2,303 ms
152,064 KB
testcase_21 AC 102 ms
156,544 KB
testcase_22 AC 2,480 ms
137,088 KB
testcase_23 AC 2,248 ms
66,176 KB
testcase_24 TLE -
testcase_25 AC 241 ms
86,784 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#line 2 "header.hpp"

//%snippet.set('header')%
//%snippet.fold()%
#ifndef HEADER_H
#define HEADER_H

// template version 2.0
using namespace std;
#include <bits/stdc++.h>

// varibable settings
const long long INF = 1e18;
template <class T> constexpr T inf = numeric_limits<T>::max() / 2.1;

#define _overload3(_1, _2, _3, name, ...) name
#define _rep(i, n) repi(i, 0, n)
#define repi(i, a, b) for (ll i = (ll)(a); i < (ll)(b); ++i)
#define rep(...) _overload3(__VA_ARGS__, repi, _rep, )(__VA_ARGS__)
#define _rrep(i, n) rrepi(i, 0, n)
#define rrepi(i, a, b) for (ll i = (ll)((b)-1); i >= (ll)(a); --i)
#define r_rep(...) _overload3(__VA_ARGS__, rrepi, _rrep, )(__VA_ARGS__)
#define each(i, a) for (auto &&i : a)
#define all(x) (x).begin(), (x).end()
#define sz(x) ((int)(x).size())
#define pb(a) push_back(a)
#define mp(a, b) make_pair(a, b)
#define mt(...) make_tuple(__VA_ARGS__)
#define ub upper_bound
#define lb lower_bound
#define lpos(A, x) (lower_bound(all(A), x) - A.begin())
#define upos(A, x) (upper_bound(all(A), x) - A.begin())
template <class T> inline void chmax(T &a, const T &b) { if ((a) < (b)) (a) = (b); }
template <class T> inline void chmin(T &a, const T &b) { if ((a) > (b)) (a) = (b); }
template <typename X, typename T> auto make_table(X x, T a) { return vector<T>(x, a); }
template <typename X, typename Y, typename Z, typename... Zs> auto make_table(X x, Y y, Z z, Zs... zs) { auto cont = make_table(y, z, zs...); return vector<decltype(cont)>(x, cont); }

#define cdiv(a, b) (((a) + (b)-1) / (b))
#define is_in(x, a, b) ((a) <= (x) && (x) < (b))
#define uni(x) sort(all(x)); x.erase(unique(all(x)), x.end())
#define slice(l, r) substr(l, r - l)

typedef long long ll;
typedef long double ld;
using vl = vector<ll>;
using vvl = vector<vl>;
using pll = pair<ll, ll>;

template <typename T>
using PQ = priority_queue<T, vector<T>, greater<T>>;
void check_input() { assert(cin.eof() == 0); int tmp; cin >> tmp; assert(cin.eof() == 1); }

#if defined(PCM) || defined(LOCAL)
#else
#define dump(...) ;
#define dump_1d(...) ;
#define dump_2d(...) ;
#define cerrendl ;
#endif

#endif /* HEADER_H */
//%snippet.end()%
#line 2 "solve.cpp"
template<class T=ll> using vec = vector<T>;
struct Fast { Fast() { std::cin.tie(0); ios::sync_with_stdio(false); } } fast;

vvl e;

// snippet:GridGraph {{{
template<class Cost=ll>
struct GridGraph {
    using Pos = ll;
    ll dim;  // 次元
    ll h, w;
    ll n;    // 頂点数
    vector<ll> size;
    vector<ll> b;
    GridGraph() {}
    GridGraph(vector<ll> _size) : size(_size) {
        dim = sz(size);
        h = size[0];
        w = size[0];
        n = 1;
        rep(i, dim) n *= size[i];
        b.resize(dim);
        b[0] = 1;
        rep(i, 1, dim) b[i] = b[i-1] * size[i-1];
    }

    Pos uid(const vector<ll>& r) {
        Pos res = 0;
        rep(i, dim) res += r[i] * b[i];
        return res;
    }
    ll operator()(const vector<ll>& r) { return uid(r); }

    vector<ll> raw(Pos x){
        vector<ll> res(dim);
        r_rep(i, dim){
            res[i] = x / b[i];
            x %= b[i];
        }
        return res;
    }

    vector<ll> get_adj(ll u){
        auto r = raw(u);
        ll i = r[0];
        ll j = r[1];
        ll remcost = r[2];
        vector<ll> adj;
        ll dx[] = {1, -1, 0, 0};
        ll dy[] = {0, 0, 1, -1};
        rep(dir, 4) {
            ll ni = i + dx[dir];
            ll nj = j + dy[dir];
            if (is_in(ni, 0, h) && is_in(nj, 0, w)){
                if (remcost >= e[ni][nj]) adj.push_back(uid({ni, nj, remcost - e[ni][nj]}));
            }
        }
        return adj;
    }

    vector<Cost> dijkstra(vector<Pos> starts) {  // 多点スタート{{{
        vector<Cost> dist(n, INF);           // 最短距離
        // PQ<pair<Cost, Pos>> pq;
        queue<pair<Cost, Pos>> pq;
        each(start, starts) {
            dist[start] = 0;
            pq.push(make_pair(0, start));
        }
        while (!pq.empty()) {
            auto cp = pq.front();
            pq.pop();
            auto [cost, u] = cp;
            auto vs = get_adj(u);
            for (const auto& v : vs) {
                Cost new_cost = cost + 1;
                if (new_cost < dist[v]) {
                    dist[v] = new_cost;
                    pq.push(make_pair(new_cost, v));
                }
            }
        }
        return dist;
    };/*}}}*/

    vector<Cost> dijkstra(Pos start) {  // 1点スタート{{{
        vector<Pos> starts = {start};
        return dijkstra(starts);
    };/*}}}*/
};
// snippet:Graph }}}

int solve() {
    ll n,v,sx,sy,gx,gy;cin>>n>>v>>sy>>sx>>gy>>gx;
    sx--;sy--;gx--;gy--;

    e = vvl(n, vl(n));
    rep(i, n)rep(j, n){
        cin>>e[i][j];
    }

    GridGraph<ll> g(vl({n, n, 2000}));
    auto d = g.dijkstra(g.uid({sx, sy, min(v, 1999LL)}));

    ll ans = INF;
    rep(r, 1, 2000){
        if (d[g.uid({gx, gy, r})] != INF){
            chmin(ans, d[g.uid({gx, gy, r})]);
        }
    }
    cout << (ans==INF ? -1 : ans) << endl;

    return 0; 
}


int main(){/*{{{*/
    solve();
    check_input();
    return 0;
}/*}}}*/
0