結果

問題 No.2328 Build Walls
ユーザー umezoumezo
提出日時 2023-05-28 16:10:48
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 312 ms / 3,000 ms
コード長 2,067 bytes
コンパイル時間 3,182 ms
コンパイル使用メモリ 216,116 KB
実行使用メモリ 72,576 KB
最終ジャッジ日時 2024-12-27 10:52:54
合計ジャッジ時間 7,320 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define ALL(v) v.begin(),v.end()
typedef long long ll;
 
#include<bits/stdc++.h>
using namespace std;

const int INF=1e9;
struct Edge{
  int to;
  int w;
  Edge(int to,ll w) : to(to),w(w) {}
};

using Graph=vector<vector<Edge>>;
using pli=pair<int,int>;

template<class T> bool chmin(T& a,T b){
  if(a>b){
    a=b;
    return true;
  }
  return false;
}

Graph G;
vector<int> dijkstra(int s){
  int n=G.size();
  vector<int> dist(n,INF);
  dist[s]=0;
  
  priority_queue<pli,vector<pli>,greater<pli>> que;
  que.push({dist[s],s});
  
  while(!que.empty()){
    int v=que.top().second;
    int d=que.top().first;
    que.pop();
    
    if(d>dist[v]) continue;
    
    for(auto e:G[v]){
      if(chmin(dist[e.to],dist[v]+e.w)){
        que.push({dist[e.to],e.to});
      }
    }
  }
  return dist;
}

int main(){
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);

  int h,w;
  cin>>h>>w;
  
  auto f=[&](int x,int y){
    return x*w+y;
  };
  
  G.resize((h-2)*w+2);
  int s=(h-2)*w,t=s+1;
  vector<vector<int>> A(h-2,vector<int>(w));
  rep(i,h-2) rep(j,w) cin>>A[i][j];
  
  rep(i,h-2){
    if(A[i][0]!=0) G[s].push_back(Edge(f(i,0),A[i][0]));
  }
  rep(i,h-2){
    if(A[i][w-1]!=0) G[f(i,w-1)].push_back(Edge(t,0));
  }                               
  rep(i,h-2) rep(j,w){
    if(A[i][j]==-1) continue;
    if(i<h-3 && A[i+1][j]!=-1){
      G[f(i,j)].push_back(Edge(f(i+1,j),A[i+1][j]));
      G[f(i+1,j)].push_back(Edge(f(i,j),A[i][j]));
    }
    if(i<h-3 && j<w-1 && A[i+1][j+1]!=-1){
      G[f(i,j)].push_back(Edge(f(i+1,j+1),A[i+1][j+1]));
      G[f(i+1,j+1)].push_back(Edge(f(i,j),A[i][j]));
    }
    if(j<w-1 && A[i][j+1]!=-1){
      G[f(i,j)].push_back(Edge(f(i,j+1),A[i][j+1]));
      G[f(i,j+1)].push_back(Edge(f(i,j),A[i][j]));
    }
    if(i && j<w-1 && A[i-1][j+1]!=-1){
      G[f(i,j)].push_back(Edge(f(i-1,j+1),A[i-1][j+1]));
      G[f(i-1,j+1)].push_back(Edge(f(i,j),A[i][j]));
    }
  }
  vector<int> dis=dijkstra(s);
  if(dis[t]==INF) cout<<-1<<endl;
  else cout<<dis[t]<<endl;
  
  return 0;
}
0