結果

問題 No.1638 Robot Maze
ユーザー ぷら
提出日時 2021-08-06 21:37:51
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 5 ms / 2,000 ms
コード長 1,933 bytes
コンパイル時間 2,137 ms
コンパイル使用メモリ 207,248 KB
最終ジャッジ日時 2025-01-23 14:49:12
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

int dx[] = {-1,1,0,0};
int dy[] = {0,0,1,-1};
int dz[] = {0,0,0,0};

bool chmin(long long &x,long long y) {
    if(x > y) {
        x = y;
        return true;
    }
    return false;
}

int main() {
    int H,W;
    cin >> H >> W >> dz[0] >> dz[1] >> dz[2] >> dz[3];
    long long K;
    int P;
    cin >> K >> P;
    int a,b,c,d;
    cin >> a >> b >> c >> d;
    a--;
    b--;
    c--;
    d--;
    vector<vector<pair<int,int>>>road(H*W);
    vector<string>S(H);
    for(int i = 0; i < H; i++) {
        cin >> S[i];
    }
    for(int i = 0; i < H; i++) {
        for(int j = 0; j < W; j++) {
            for(int k = 0; k < 4; k++) {
                int nx = i+dx[k];
                int ny = j+dy[k];
                if(nx >= 0 && nx < H && ny >= 0 && ny < W) {
                    if(S[nx][ny] == '#') {
                        continue;
                    }
                    if(S[nx][ny] == '@') {
                        road[i*W+j].push_back({nx*W+ny,P+dz[k]});
                    }
                    else {
                        road[i*W+j].push_back({nx*W+ny,dz[k]});
                    }
                }
            }
        }
    }
    vector<long long>dist(H*W,1001001001001001001);
    dist[a*W+b] = 0;
    priority_queue<pair<long long,int>,vector<pair<long long,int>>,greater<pair<long long,int>>>que;
    que.push({0,a*W+b});
    while (!que.empty()) {
        pair<long long,int>x = que.top();
        que.pop();
        if(dist[x.second] != x.first) {
            continue;
        }
        for(int i = 0; i < road[x.second].size(); i++) {
            if(chmin(dist[road[x.second][i].first],road[x.second][i].second+x.first)) {
                que.push({dist[road[x.second][i].first],road[x.second][i].first});
            }
        }
    }
    if(dist[c*W+d] <= K) {
        cout << "Yes" << endl;
    }
    else {
        cout << "No" << endl;
    }
}
0