結果

問題 No.20 砂漠のオアシス
ユーザー sarashinsarashin
提出日時 2019-07-01 15:15:21
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 33 ms / 5,000 ms
コード長 2,008 bytes
コンパイル時間 2,124 ms
コンパイル使用メモリ 181,376 KB
実行使用メモリ 7,936 KB
最終ジャッジ日時 2024-07-16 02:34:59
合計ジャッジ時間 2,878 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 3 ms
6,940 KB
testcase_04 AC 4 ms
6,940 KB
testcase_05 AC 27 ms
7,320 KB
testcase_06 AC 33 ms
7,808 KB
testcase_07 AC 33 ms
7,808 KB
testcase_08 AC 33 ms
7,936 KB
testcase_09 AC 33 ms
7,808 KB
testcase_10 AC 2 ms
6,944 KB
testcase_11 AC 2 ms
6,940 KB
testcase_12 AC 3 ms
6,944 KB
testcase_13 AC 3 ms
6,944 KB
testcase_14 AC 4 ms
6,940 KB
testcase_15 AC 3 ms
6,940 KB
testcase_16 AC 7 ms
6,940 KB
testcase_17 AC 5 ms
6,940 KB
testcase_18 AC 6 ms
6,944 KB
testcase_19 AC 8 ms
6,940 KB
testcase_20 AC 3 ms
6,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#define ll long long
#define INF 1e9
#define LINF 1e18
#define rep(i,n) for(int i=0;i<n;++i)
using namespace std;
typedef pair<ll,ll> P;
int dx[4] = {0, 0, 1, -1};
int dy[4] = {1, -1, 0, 0};

typedef struct Edge {
    ll to,cost;
} Edge;

vector<vector<Edge>> g;
int L[222][222];


void print(const vector<ll> &dp, int n) {
    rep(i,n) {
        rep(j,n) {
            cout << i*n+j << " ";
        }
        cout << endl;
    }
}

void make_graph(int n, int m) {
    rep(i,n) {
        rep(j,m) {
            rep(k,4) {
                int now = i*n+j;
                int nx = i+dx[k]; int ny = j+dy[k];
                if (-1 < nx && nx < n && -1 < ny && ny < m) {
                    // printf("now = %d, nx = %d, ny = %d\n", now,nx,ny);
                    int cost = L[nx][ny];
                    int to = nx*n+ny;
                    g[now].push_back(Edge{to,cost});
                } 
            }
        }
    }
}

void dijkstra(vector<ll> &dp, int s) {
    priority_queue<P, vector<P>, greater<P>> pq;
    pq.push({0,s});
    dp[s] = 0;

    while (!pq.empty()) {
        P p = pq.top(); pq.pop();
        int v = p.second;
        for (auto &&e: g[v]) {
            if (dp[e.to] > dp[v]+e.cost) {
                dp[e.to] = dp[v]+e.cost;
                pq.push({dp[e.to], e.to});
            }
        }
    }
}

int main() {
    int n,v,ox,oy;
    cin >> n >> v >> ox >> oy;
    --ox, --oy;
    g.resize(n*n);
    rep(i,n) rep(j,n) cin >> L[i][j];
    make_graph(n,n);

    vector<ll> dp(n*n, INF);
    dijkstra(dp,0);
    int goal = n*n-1;
    if (v-dp[goal] > 0) {
        cout << "YES" << endl;
        return 0;
    }

    if (ox == -1 && oy == -1) {
        cout << "NO" << endl;
        return 0;
    }

    int oasis = oy*n+ox;
    int oasis_v = v-dp[oasis];
    oasis_v *= 2;
    dp.assign(n*n,INF);
    dijkstra(dp,oasis);
    
    if (oasis_v-dp[goal] > 0) {
        cout << "YES" << endl;
        return 0;
    }

    cout << "NO" << endl;
    return 0;
}
0