結果
問題 | No.867 避難経路 |
ユーザー |
![]() |
提出日時 | 2019-08-16 23:28:22 |
言語 | C++11 (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 5,555 ms / 6,000 ms |
コード長 | 2,180 bytes |
コンパイル時間 | 1,570 ms |
コンパイル使用メモリ | 169,588 KB |
実行使用メモリ | 151,040 KB |
最終ジャッジ日時 | 2024-09-22 22:52:44 |
合計ジャッジ時間 | 139,571 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include <bits/stdc++.h>using namespace std;typedef long long ll;typedef unsigned long long ull;typedef pair<ll, ll> P;#define fi first#define se second#define repl(i,a,b) for(ll i=(ll)(a);i<(ll)(b);i++)#define rep(i,n) repl(i,0,n)#define all(x) (x).begin(),(x).end()#define dbg(x) cout<<#x"="<<x<<endl#define mmax(x,y) (x>y?x:y)#define mmin(x,y) (x<y?x:y)#define maxch(x,y) x=mmax(x,y)#define minch(x,y) x=mmin(x,y)#define uni(x) x.erase(unique(all(x)),x.end())#define exist(x,y) (find(all(x),y)!=x.end())#define bcnt __builtin_popcountll#define INF 1e16#define mod 1000000007struct state{ll i,j,d;};bool operator<(const state& a,const state& b){return a.d > b.d;}struct state2{ll i,j;P p;};bool operator<(const state2& a,const state2& b){return a.p > b.p;}ll H,W;ll gi,gj;ll A[250][250];ll Q;ll dist[300][250][250];P dist2[250][250];ll di[]={-1,0,+1,0};ll dj[]={0,-1,0,+1};const ll Kmax=300;int main(){cin.tie(0);ios::sync_with_stdio(false);cin>>H>>W>>gi>>gj;gi--;gj--;rep(i,H)rep(j,W){cin>>A[i][j];}repl(k,1,Kmax){rep(i,H)rep(j,W)dist[k][i][j]=INF;priority_queue<state> que;que.push((state){gi,gj,k*k+A[gi][gj]});while(que.size()){state s=que.top(); que.pop();if(dist[k][s.i][s.j]!=INF)continue;dist[k][s.i][s.j]=s.d;rep(dir,4){ll ni=s.i+di[dir],nj=s.j+dj[dir];if(ni<0||ni>=H||nj<0||nj>=W)continue;que.push((state){ni,nj,s.d+k*k+A[ni][nj]});}}}{rep(i,H)rep(j,W)dist2[i][j]=P(INF,INF);priority_queue<state2> que2;que2.push((state2){gi,gj,P(1,A[gi][gj])});while(que2.size()){state2 s=que2.top(); que2.pop();if(dist2[s.i][s.j]!=P(INF,INF))continue;dist2[s.i][s.j]=s.p;rep(dir,4){ll ni=s.i+di[dir],nj=s.j+dj[dir];if(ni<0||ni>=H||nj<0||nj>=W)continue;que2.push((state2){ni,nj,P(s.p.fi+1,s.p.se+A[ni][nj])});}}}cin>>Q;rep(_,Q){ll x,y,k;cin>>x>>y>>k;x--;y--;if(k<Kmax) cout<<dist[k][x][y]<<'\n';else{P p=dist2[x][y];cout<<p.fi*k*k+p.se<<'\n';}}return 0;}