結果

問題 No.20 砂漠のオアシス
コンテスト
ユーザー monnu
提出日時 2021-07-16 19:59:26
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 17 ms / 5,000 ms
コード長 2,047 bytes
コンパイル時間 1,489 ms
コンパイル使用メモリ 177,420 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-06 06:36:51
合計ジャッジ時間 2,203 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;
//#include <atcoder/all>
//using namespace atcoder;
using ll=long long;
using Graph=vector<vector<int>>;
#define MAX 1000000
#define MOD 1000000007
//#define MOD 998244353
#define INF 1000000000
//#define INF 1000000000000000000

int main(){
  int N,V,Ox,Oy;
  cin>>N>>V>>Ox>>Oy;
  vector<vector<int>> L(N,vector<int>(N));
  for(int i=0;i<N;i++){
    for(int j=0;j<N;j++){
      cin>>L[i][j];
    }
  }

  priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>> pq;
  vector<vector<int>> dist(N,vector<int>(N,INF));
  dist[0][0]=0;
  pq.push(make_pair(dist[0][0],0));
  int di[4]={1,0,-1,0};
  int dj[4]={0,1,0,-1};
  while(!pq.empty()){
    int v=pq.top().second;
    int d=pq.top().first;
    pq.pop();
    int i=v/N;
    int j=v%N;
    if(dist[i][j]<d){
      continue;
    }
    for(int k=0;k<4;k++){
      int ni=i+di[k];
      int nj=j+dj[k];
      if(ni<0||nj<0||ni>=N||nj>=N){
        continue;
      }
      if(dist[ni][nj]>dist[i][j]+L[ni][nj]){
        dist[ni][nj]=dist[i][j]+L[ni][nj];
        pq.push(make_pair(dist[ni][nj],ni*N+nj));
      }
    }
  }
  if(dist[N-1][N-1]<V){
    cout<<"YES"<<endl;
    return 0;
  }
  if(Ox==0&&Oy==0){
    cout<<"NO"<<endl;
    return 0;
  }
  swap(Ox,Oy);
  Ox--;
  Oy--;
  V-=dist[Ox][Oy];
  if(V<=0){
    cout<<"NO"<<endl;
    return 0;
  }
  V*=2;

  for(int i=0;i<N;i++){
    for(int j=0;j<N;j++){
      dist[i][j]=INF;
    }
  }
  dist[Ox][Oy]=0;
  pq.push(make_pair(dist[Ox][Oy],Ox*N+Oy));
  while(!pq.empty()){
    int v=pq.top().second;
    int d=pq.top().first;
    pq.pop();
    int i=v/N;
    int j=v%N;
    if(dist[i][j]<d){
      continue;
    }
    for(int k=0;k<4;k++){
      int ni=i+di[k];
      int nj=j+dj[k];
      if(ni<0||nj<0||ni>=N||nj>=N){
        continue;
      }
      if(dist[ni][nj]>dist[i][j]+L[ni][nj]){
        dist[ni][nj]=dist[i][j]+L[ni][nj];
        pq.push(make_pair(dist[ni][nj],ni*N+nj));
      }
    }
  }
  if(dist[N-1][N-1]<V){
    cout<<"YES"<<endl;
  }else{
    cout<<"NO"<<endl;
  }
}
0