結果

問題 No.867 避難経路
コンテスト
ユーザー DAyamaCTF
提出日時 2019-08-16 23:36:51
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 4,648 ms / 6,000 ms
コード長 2,180 bytes
コンパイル時間 1,475 ms
コンパイル使用メモリ 170,632 KB
実行使用メモリ 129,444 KB
最終ジャッジ日時 2024-09-22 23:27:49
合計ジャッジ時間 119,850 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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 1000000007

struct 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=251;

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;
}
0