結果

問題 No.867 避難経路
コンテスト
ユーザー b312546656
提出日時 2019-08-16 22:24:54
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 3,254 bytes
コンパイル時間 1,479 ms
コンパイル使用メモリ 94,916 KB
実行使用メモリ 44,672 KB
最終ジャッジ日時 2024-09-22 18:14:08
合計ジャッジ時間 78,906 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22 WA * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<algorithm>
#include<vector>
#include<string>
#include<set>
#include<queue>
#include<stack>
#include<bitset>
#include<functional>
#include<map>
#include<unordered_set>
using namespace std;
/*int p = 998244353;*/
int p = 1000000007;
#define vel vector<int>
#define vvel vector<vel>
#define rep(i,n) for(int i=0;i<n;i++)
#define sor(v) sort(v.begin(),v.end())
#define mmax(a,b) a=max(a,b)
#define mmin(a,b) a=min(a,b)
#define mkp make_pair
#define pin pair<int,int>
#define qin pair<int,pin>
#define V vector
#define Endl endl
#define veb vector<bool>
#define sq(a) (a)*(a)
#define rev(s) reverse(s.begin(),s.end())
#define end_pr(s) cout << s <<endl;return 0
#define itn int
int RE() {
	vel v(3, 2);
	return v.at(4);
}
int ru(int a, int r) {
	if (r == 0) { return 1; }
	int ans = ru(a, r / 2);
	ans *= ans; ans %= p;
	if (r % 2 == 1) { ans *= a; }
	return ans % p;
}
int inv(int a) {
	return ru(a,p - 2);
}
vel dis(int mid1, vvel &way) {
	int n = way.size();
	vel dist(n, -1); dist[mid1] = 0;
	queue<int> q;
	q.push(mid1);
	while (!q.empty()) {
		int st = q.front(); q.pop();
		rep(i, way[st].size()) {
			int to = way[st][i];
			if (dist[to] == -1) {
				dist[to] = dist[st] + 1;
				q.push(to);
			}
		}
	}
	return dist;
}
pin most_far(int now, int n, vvel &way) {
	vel dist1 = dis(now, way);
	pin ans = mkp(-1, 0);
	rep(i, n) {
		if (dist1[i] > ans.first) { ans = mkp(dist1[i], i); }
	}
	return ans;
}

template<typename T>
V<T> uni(V<T> &v) {
	if (v.size() == 0) { return v; }
	sor(v);
	V<T> ans(1, v[0]);
	for (int i = 1; i < v.size(); i++) {
		if (v[i] != v[i - 1]) { ans.push_back(v[i]); }
	}
	v = ans;
	return v;
}
int s_gcd(int a, int b) {
	if (b == 0) { return a; }
	return s_gcd(b, a%b);
}
int gcd(int a, int b) {
	a = abs(a);
	b = abs(b);
	if (a < b) { swap(a, b); }
	return s_gcd(a, b);
}
int solve(int sum, int lim) {
	if (sum <= lim) { return sum - 1; }
	if (sum > lim * 2) { return 0; }
	return lim * 2 - sum+1;
}
vel dx{ 1,-1,0,0 };
vel dy{ 0,0,1,-1 };
vvel dijk(vvel &a, int h, int w, int k,int gx,int gy) {
	priority_queue<qin,vector<qin>,greater<qin>> pq;
	vvel dist(h, vel(w, -1));
	pq.push(mkp(0, mkp(gx, gy)));
	while (!pq.empty()) {
		qin sm = pq.top(); pq.pop();
		pin pot = sm.second;
		int x = pot.first;
		int y = pot.second;
		int dis1 = sm.first;
		if (dist[x][y] == -1) {
			dist[x][y] = dis1;
			rep(i, 4) {
				int mx = x + dx[i];
				int my = y + dy[i];
				if (0 <= mx and mx < h and 0 <= my and my < w) {
					if (dist[mx][my] == -1) {
						pq.push(mkp(dis1 + k * k + a[mx][my], mkp(mx, my)));
					}
				}
			}
		}
	}
	rep(i, h) { rep(j, w) { dist[i][j] += (k * k + a[gx][gy]); } }
	return dist;
}
int upper = 160;
signed main() {
	int h, w; cin >> h >> w;
	int gx, gy; cin >> gx >> gy;
	gx--; gy--;
	vvel a(h, vel(w));
	rep(i, h) { rep(j, w) { cin >> a[i][j]; } }
	V<vvel> dist(upper+1, vvel(h, vel(w)));
	for (int k = 1; k <= upper; k++) {
		dist[k] = dijk(a, h, w, k,gx,gy);
	}
	int q; cin >> q;
	rep(i, q) {
		int x, y;long long k; cin >> x >> y >> k; x--; y--;
		if (k <= upper) {
			cout << dist[k][x][y] << endl;
		}
		else {
			int qans = dist[k][x][y];
			int c_dis = abs(x - gx) + abs(y - gy);
			c_dis++;
			long long mul = (k*k) - upper * upper;
			qans += c_dis + mul;
		}
	}
}
0