結果

問題 No.867 避難経路
コンテスト
ユーザー e869120
提出日時 2019-08-15 11:18:47
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,558 bytes
コンパイル時間 911 ms
コンパイル使用メモリ 85,980 KB
実行使用メモリ 132,352 KB
最終ジャッジ日時 2024-09-19 15:06:33
合計ジャッジ時間 65,448 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22 WA * 19
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <functional>
using namespace std;
#pragma warning (disable: 4996)

long long H, W, Q, gx, gy, A[259][259];
long long val[259][259][259], dist[259][259];
priority_queue<tuple<long long, int, int>, vector<tuple<long long, int, int>>, greater<tuple<long long, int, int>>> Que;

void dijkstra(long long r) {
	for (int i = 1; i <= H; i++) {
		for (int j = 1; j <= W; j++) dist[i][j] = (1LL << 62);
	}
	dist[gx][gy] = 1LL * (r*r + A[gx][gy]);
	Que.push(make_tuple(dist[gx][gy], gx, gy));

	int dx[4] = { 1, 0, -1, 0 }, dy[4] = { 0,1, 0, -1 };

	while (!Que.empty()) {
		tuple<long long, int, int> tup = Que.top(); Que.pop();
		int cx = get<1>(tup), cy = get<2>(tup);
		for (int i = 0; i < 4; i++) {
			int ex = cx + dx[i], ey = cy + dy[i];
			if (ex <= 0 || ey <= 0 || ex > H || ey > W) continue;

			long long cost = r * r + A[ex][ey];
			if (dist[ex][ey] > dist[cx][cy] + 1LL * cost) {
				dist[ex][ey] = dist[cx][cy] + 1LL * cost;
				Que.push(make_tuple(dist[ex][ey], ex, ey));
			}
		}
	}
}

int main() {
	scanf("%lld%lld%lld%lld", &H, &W, &gx, &gy);
	for (int i = 1; i <= H; i++) {
		for (int j = 1; j <= W; j++) scanf("%lld", &A[i][j]);
	}
	for (int i = 1; i <= 250; i++) {
		dijkstra(i);
		for (int j = 1; j <= H; j++) {
			for (int k = 1; k <= W; k++) val[i][j][k] = dist[j][k];
		}
	}
	scanf("%lld", &Q);
	for (int i = 1; i <= Q; i++) {
		long long cx, cy, k;
		scanf("%lld%lld%lld", &cx, &cy, &k); k = min(k, 250LL);
		printf("%lld\n", val[k][cx][cy]);
	}
	return 0;
}
0