QUPC2018 C - Ito Campus
解法
各イノシシからの距離を別個で計算するのではなくまとめて計算することができる。グラフの仮想頂点としてイノシシまでの距離が0の辺を張った頂点を追加してその頂点からbfsをスタートするイメージ。イノシシからの距離をbfsで求めるのがO(HW)、うしくんがゴールまでの最短経路を求めるのがO(HW)なので解ける。
#include <bits/stdc++.h> using namespace std; using ll = long long; // #define int ll using PII = pair<int, int>; #define FOR(i, a, n) for (ll i = (ll)a; i < (ll)n; ++i) #define REP(i, n) FOR(i, 0, n) #define ALL(x) x.begin(), x.end() template<typename T> T &chmin(T &a, const T &b) { return a = min(a, b); } template<typename T> T &chmax(T &a, const T &b) { return a = max(a, b); } template<typename T> bool IN(T a, T b, T x) { return a<=x&&x<b; } template<typename T> T ceil(T a, T b) { return a/b + !!(a%b); } template<typename T> vector<T> make_v(size_t a) { return vector<T>(a); } template<typename T,typename... Ts> auto make_v(size_t a,Ts... ts) { return vector<decltype(make_v<T>(ts...))>(a,make_v<T>(ts...)); } template<typename T,typename V> typename enable_if<is_class<T>::value==0>::type fill_v(T &t, const V &v) { t=v; } template<typename T,typename V> typename enable_if<is_class<T>::value!=0>::type fill_v(T &t, const V &v ) { for(auto &e:t) fill_v(e,v); } template<class S,class T> ostream &operator <<(ostream& out,const pair<S,T>& a){ out<<'('<<a.first<<','<<a.second<<')'; return out; } template<typename T> istream& operator >> (istream& is, vector<T>& vec){ for(T& x: vec) {is >> x;} return is; } template<class T> ostream &operator <<(ostream& out,const vector<T>& a){ out<<'['; for(T i: a) {out<<i<<',';} out<<']'; return out; } int dx[] = {0, 1, 0, -1}, dy[] = {1, 0, -1, 0}; // DRUL const int INF = 1<<30; const ll LLINF = 1LL<<60; const int MOD = 1000000007; signed main(void) { cin.tie(0); ios::sync_with_stdio(false); int h, w, x; cin >> h >> w >> x; vector<string> v(h); REP(i, h) cin >> v[i]; int sx, sy, gx, gy; vector<int> ix, iy; REP(i, h) REP(j, w) { if(v[i][j] == '@') { ix.push_back(j); iy.push_back(i); } if(v[i][j] == 'S') { sx = j, sy = i; } if(v[i][j] == 'G') { gx = j, gy = i; } } auto d1 = make_v<int>(h,w); fill_v(d1, INF); queue<PII> que; REP(i, ix.size()) { que.push({ix[i], iy[i]}); d1[iy[i]][ix[i]] = 0; } while(que.size()) { PII p = que.front(); que.pop(); REP(i, 4) { int nx = p.first + dx[i], ny = p.second + dy[i]; if(!IN(0,h,ny) || !IN(0,w,nx) || v[ny][nx]=='#') continue; if(d1[ny][nx] > d1[p.second][p.first] + 1) { d1[ny][nx] = d1[p.second][p.first] + 1; que.push({nx, ny}); } } } auto d2 = make_v<int>(h,w); fill_v(d2, INF); que.push({sx, sy}); d2[sy][sx] = 0; while(que.size()) { PII p = que.front(); que.pop(); REP(i, 4) { int nx = p.first + dx[i], ny = p.second + dy[i]; if(!IN(0,h,ny) || !IN(0,w,nx) || v[ny][nx]=='#' || d1[ny][nx] <= x) continue; if(d2[ny][nx] > d2[p.second][p.first] + 1) { d2[ny][nx] = d2[p.second][p.first] + 1; que.push({nx, ny}); } } } if(d2[gy][gx] == INF) cout << -1 << endl; else cout << d2[gy][gx] << endl; return 0; }