結果

問題 No.867 避難経路
ユーザー ゆにぽけゆにぽけ
提出日時 2024-03-11 15:06:50
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2,805 ms / 6,000 ms
コード長 3,002 bytes
コンパイル時間 1,677 ms
コンパイル使用メモリ 143,148 KB
実行使用メモリ 61,312 KB
最終ジャッジ日時 2024-09-29 22:01:53
合計ジャッジ時間 75,968 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
5,248 KB
testcase_01 AC 3 ms
5,248 KB
testcase_02 AC 3 ms
5,248 KB
testcase_03 AC 3 ms
5,248 KB
testcase_04 AC 3 ms
5,248 KB
testcase_05 AC 3 ms
5,248 KB
testcase_06 AC 3 ms
5,248 KB
testcase_07 AC 3 ms
5,248 KB
testcase_08 AC 3 ms
5,248 KB
testcase_09 AC 3 ms
5,248 KB
testcase_10 AC 3 ms
5,248 KB
testcase_11 AC 3 ms
5,248 KB
testcase_12 AC 3 ms
5,248 KB
testcase_13 AC 2 ms
5,248 KB
testcase_14 AC 3 ms
5,248 KB
testcase_15 AC 2,721 ms
61,184 KB
testcase_16 AC 2,607 ms
61,184 KB
testcase_17 AC 2,751 ms
61,056 KB
testcase_18 AC 2,738 ms
61,056 KB
testcase_19 AC 2,624 ms
61,056 KB
testcase_20 AC 2,508 ms
61,184 KB
testcase_21 AC 2,544 ms
61,056 KB
testcase_22 AC 2,448 ms
61,056 KB
testcase_23 AC 2,717 ms
61,184 KB
testcase_24 AC 2,685 ms
61,056 KB
testcase_25 AC 2,360 ms
61,312 KB
testcase_26 AC 2,585 ms
61,184 KB
testcase_27 AC 2,734 ms
61,184 KB
testcase_28 AC 2,731 ms
61,184 KB
testcase_29 AC 2,805 ms
61,184 KB
testcase_30 AC 2,570 ms
61,056 KB
testcase_31 AC 2,725 ms
61,056 KB
testcase_32 AC 2,457 ms
61,056 KB
testcase_33 AC 2,467 ms
61,056 KB
testcase_34 AC 2,327 ms
60,672 KB
testcase_35 AC 2,650 ms
60,672 KB
testcase_36 AC 2,513 ms
59,776 KB
testcase_37 AC 2,275 ms
56,960 KB
testcase_38 AC 2,336 ms
57,088 KB
testcase_39 AC 2,213 ms
57,088 KB
testcase_40 AC 2,365 ms
57,088 KB
testcase_41 AC 2 ms
5,248 KB
testcase_42 AC 3 ms
5,248 KB
testcase_43 AC 3 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <array>
#include <iterator>
#include <string>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <cassert>
#include <cmath>
#include <ctime>
#include <iomanip>
#include <numeric>
#include <stack>
#include <queue>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <bitset>
#include <random>
#include <utility>
#include <functional>
using namespace std;
void Main()
{
	int H,W,gi,gj;
	cin >> H >> W >> gi >> gj;
	gi--; gj--;
	vector<vector<int>> A(H,vector<int>(W));
	for(int i = 0;i < H;i++)
	{
		for(int j = 0;j < W;j++)
		{
			cin >> A[i][j];
		}
	}
	vector<vector<int>> cost(H,vector<int>(W,(int)1e9));
	const int d[5] = {0,1,0,-1};
	cost[gi][gj] = A[gi][gj];
	auto dfs = [&](auto dfs,int i,int j) -> int 
	{
		if(cost[i][j] < (int)1e9)
		{
			return cost[i][j];
		}
		int res = (int)1e9;
		if(i < gi)
		{
			res = min(res,dfs(dfs,i + 1,j) + A[i][j]);
		}
		if(i > gi)
		{
			res = min(res,dfs(dfs,i - 1,j) + A[i][j]);
		}
		if(j < gj)
		{
			res = min(res,dfs(dfs,i,j + 1) + A[i][j]);
		}
		if(j > gj)
		{
			res = min(res,dfs(dfs,i,j - 1) + A[i][j]);
		}
		return cost[i][j] = res;
	};
	for(int i = 0;i < H;i++)
	{
		for(int j = 0;j < W;j++)
		{
			dfs(dfs,i,j);
		}
	}
	/* { */
	/* 	queue<pair<int,int>> Q; */
	/* 	Q.push(make_pair(gi,gj)); */
	/* 	while(!Q.empty()) */
	/* 	{ */
	/* 		auto [i,j] = Q.front(); */
	/* 		Q.pop(); */
	/* 		for(int k = 0;k < 4;k++) */
	/* 		{ */
	/* 			int ni = i + d[k]; */
	/* 			int nj = j + d[k + 1]; */
	/* 			if(ni < 0 || nj < 0 || ni >= H || nj >= W) */
	/* 			{ */
	/* 				continue; */
	/* 			} */
	/* 			if(cost[ni][nj] == (int)1e9) */
	/* 			{ */
	/* 				Q.push(make_pair(ni,nj)); */
	/* 			} */
	/* 			cost[ni][nj] = min(cost[ni][nj],cost[i][j] + A[ni][nj]); */
	/* 		} */
	/* 	} */
	/* } */
	const int B = 223;
	vector<vector<vector<int>>> dp(H,vector<vector<int>>(W,vector<int>(B,(int)1e9)));
	for(int k = 1;k < B;k++)
	{
		dp[gi][gj][k] = k * k + A[gi][gj];
		priority_queue<pair<int,pair<int,int>>> Q;
		Q.push(make_pair(-dp[gi][gj][k],make_pair(gi,gj)));
		while(!Q.empty())
		{
			int cur = - Q.top().first;
			auto [i,j] = Q.top().second;
			Q.pop();
			if(dp[i][j][k] < cur)
			{
				continue;
			}
			for(int r = 0;r < 4;r++)
			{
				int ni = i + d[r];
				int nj = j + d[r + 1];
				if(ni < 0 || nj < 0 || ni >= H || nj >= W)
				{
					continue;
				}
				if(dp[ni][nj][k] > dp[i][j][k] + k * k + A[ni][nj])
				{
					dp[ni][nj][k] = dp[i][j][k] + k * k + A[ni][nj];
					Q.push(make_pair(-dp[ni][nj][k],make_pair(ni,nj)));
				}
			}
		}
	}
	int Q;
	cin >> Q;
	for(;Q--;)
	{
		int x,y,k;
		cin >> x >> y >> k;
		x--; y--;
		long long ans = 0;
		if(k >= B)
		{
			ans = (long long) k * k * (abs(x - gi) + abs(y - gj) + 1) + cost[x][y];
		}
		else
		{
			ans = dp[x][y][k];
		}
		cout << ans << "\n";
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int tt = 1;
	/* cin >> tt; */
	while(tt--) Main();
}
0