#686453 (C++14) No.1638 Robot Maze

提出ソース

結果

問題	No.1638 Robot Maze
ユーザー	ecottea
提出日時	2021-08-06 21:51:02
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	WA
実行時間	-
コード長	9,110 bytes
コンパイル時間	4,105 ms
コンパイル使用メモリ	245,408 KB
実行使用メモリ	6,948 KB
最終ジャッジ日時	2024-09-17 01:45:40
合計ジャッジ時間	5,216 ms
ジャッジサーバーID （参考情報）	judge6 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	2 ms 6,944 KB
testcase_02	AC	2 ms 6,944 KB
testcase_03	AC	8 ms 6,940 KB
testcase_04	AC	2 ms 6,944 KB
testcase_05	WA	-
testcase_06	AC	2 ms 6,944 KB
testcase_07	AC	2 ms 6,940 KB
testcase_08	AC	2 ms 6,944 KB
testcase_09	AC	2 ms 6,940 KB
testcase_10	AC	2 ms 6,940 KB
testcase_11	AC	2 ms 6,944 KB
testcase_12	AC	1 ms 6,940 KB
testcase_13	AC	2 ms 6,940 KB
testcase_14	WA	-
testcase_15	AC	4 ms 6,944 KB
testcase_16	AC	3 ms 6,940 KB
testcase_17	AC	4 ms 6,944 KB
testcase_18	AC	3 ms 6,940 KB
testcase_19	AC	3 ms 6,940 KB
testcase_20	AC	4 ms 6,944 KB
testcase_21	AC	3 ms 6,944 KB
testcase_22	AC	3 ms 6,940 KB
testcase_23	AC	3 ms 6,940 KB
testcase_24	AC	3 ms 6,940 KB
testcase_25	AC	3 ms 6,944 KB
testcase_26	AC	3 ms 6,940 KB
testcase_27	AC	3 ms 6,944 KB
testcase_28	AC	3 ms 6,940 KB
testcase_29	WA	-
testcase_30	AC	2 ms 6,944 KB
testcase_31	AC	2 ms 6,940 KB
testcase_32	AC	5 ms 6,940 KB
testcase_33	AC	6 ms 6,944 KB
testcase_34	AC	7 ms 6,940 KB
testcase_35	AC	7 ms 6,944 KB
testcase_36	AC	7 ms 6,944 KB
testcase_37	AC	6 ms 6,944 KB
testcase_38	AC	6 ms 6,940 KB
testcase_39	AC	7 ms 6,940 KB
testcase_40	AC	7 ms 6,944 KB
testcase_41	AC	6 ms 6,944 KB
testcase_42	AC	6 ms 6,940 KB
testcase_43	AC	7 ms 6,940 KB
testcase_44	AC	7 ms 6,944 KB
testcase_45	AC	7 ms 6,948 KB
testcase_46	AC	7 ms 6,944 KB
testcase_47	AC	6 ms 6,940 KB
testcase_48	AC	7 ms 6,944 KB
testcase_49	AC	5 ms 6,940 KB
testcase_50	AC	6 ms 6,940 KB
testcase_51	AC	7 ms 6,944 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#ifndef HIDDEN_IN_VISUAL_STUDIO // 無意味．折りたたむのが目的．

// 警告の抑制
#define _CRT_SECURE_NO_WARNINGS

// 使えるライブラリの読み込み
#include <bits/stdc++.h>
#include <functional> // function
using namespace std;

// 型名の短縮
using ll = long long;           // -2^63 ～ 2^63 = 9   * 10^18
using ull = unsigned long long; //     0 ～ 2^64 = 1.8 * 10^19
using uint = unsigned int;      //     0 ～ 2^32 = 4   * 10^9
using pii = pair<int, int>;
using pll = pair<ll, ll>;

// 定数の定義
const double PI = 3.141592653589793238462643383279; // 円周率
const double DEG = PI / 180.0; // θ [deg] = θ * DEG [rad]
const vector<int> dx4 = { 1, 0, -1, 0 }; // 4 近傍
const vector<int> dy4 = { 0, 1, 0, -1 };
const vector<int> dx8 = { 1, 1, 0, -1, -1, -1, 0, 1 }; // 8 近傍
const vector<int> dy8 = { 0, 1, 1, 1, 0, -1, -1, -1 };
const ll INFL = (ll)1e18;
const int INF = (int)1e9;
const double EPS = 1e-10; // 許容誤差に応じて調整

// 汎用マクロの定義
#define all(a) (a).begin(), (a).end()
#define sz(x) ((int)(x).size())
#define rep(i, n) for(int i = 0, i##_len = ((int)(n)); i < i##_len; ++i) // 0 から n-1 まで昇順
#define repi(i, s, t) for(int i = (int)(s), i##_end = ((int)(t)); i <= i##_end; ++i) // s から t まで昇順
#define repir(i, s, t) for(int i = (int)(s), i##_end = ((int)(t)); i >= i##_end; --i) // s から t まで降順
#define repe(i, a) for(const auto& i : (a)) // a の全要素
#define repb(set, d) for(int set = 0; set < (1 << (int)(d)); ++set) // d ビット全探索（昇順）
#define repbm(mid, set, d) for(int mid = set; mid < (1 << (int)(d)); mid = (mid + 1) | set) // set を含む部分集合の全探索（昇順）
#define repbs(sub, set) for (int sub = set, bsub = 1; bsub > 0; bsub = sub, sub = (sub - 1) & set) // set の部分集合の全探索（降順）
#define repbc(set, k, d) for (int set = (1 << k) - 1, lb, nx; set < (1 << n); lb = set & -set, nx = set + lb, set = (((set & ~nx) / lb) >> 1) | nx) // 大きさ k の部分集合の全探索
#define repp(a) sort(all(a)); for(bool a##_perm = true; a##_perm; a##_perm = next_permutation(all(a))) // a の順列全て（昇順）
#define repit(it, a) for(auto it = (a).begin(); it != (a).end(); it++) // イテレータを回す（昇順）
#define Yes(b) if(b){cout << "Yes" << endl;}else{cout << "No" << endl;}
#define Tak(b) if(b){cout << "Takahashi" << endl;}else{cout << "Aoki" << endl;}

// 汎用関数の定義
inline ll pow(ll n, int k) { ll v = 1; rep(i, k) v *= n; return v; } // 工夫が必要なほど k が大きかったらどうせオーバーフローするからこれでいい
inline ll pow(int n, int k) { ll v = 1; rep(i, k) v *= (ll)n; return v; }
template <class T> inline bool chmax(T& M, const T& x) { if (M < x) { M = x; return true; } return false; } // 最大値を更新（更新されたら true を返す）
template <class T> inline bool chmin(T& m, const T& x) { if (m > x) { m = x; return true; } return false; } // 最小値を更新（更新されたら true を返す）

// 入出力用の >>, << のオーバーロード
template <class T, class U> istream& operator>> (istream& is, pair<T, U>& p) { is >> p.first >> p.second; return is; } // pair の入力用
template <class T, class U> ostream& operator<< (ostream& os, const pair<T, U>& p) { os << "(" << p.first << "," << p.second << ")"; return os; } // pair の出力用
template <class T, class U, class V> istream& operator>> (istream& is, tuple<T, U, V>& t) { is >> get<0>(t) >> get<1>(t) >> get<2>(t); return is; } // tuple の入力用
template <class T, class U, class V> ostream& operator<< (ostream& os, const tuple<T, U, V>& t) { os << "(" << get<0>(t) << "," << get<1>(t) << "," << get<2>(t) << ")"; return os; } // tuple の出力用
template <class T, class U, class V, class W> istream& operator>> (istream& is, tuple<T, U, V, W>& t) { is >> get<0>(t) >> get<1>(t) >> get<2>(t) >> get<3>(t); return is; } // tuple の入力用
template <class T, class U, class V, class W> ostream& operator<< (ostream& os, const tuple<T, U, V, W>& t) { os << "(" << get<0>(t) << "," << get<1>(t) << "," << get<2>(t) << "," << get<3>(t) << ")"; return os; } // tuple の出力用
template <class T> istream& operator>> (istream& is, vector<T>& v) { rep(i, sz(v)) is >> v[i]; return is; } // vector の入力用
template <class T> ostream& operator<< (ostream& os, const vector<T>& v) { rep(i, sz(v)) os << v[i] << " "; return os; } // vector の出力用
template <class T> ostream& operator<< (ostream& os, const set<T>& s) { repe(x, s) os << x << " "; return os; } // set の出力用

// 手元環境（Visual Studio）
#ifdef _MSC_VER
#define popcount (int)__popcnt // 全ビットにおける 1 の個数
#define popcountll (int)__popcnt64
inline int ctz(uint n) { unsigned long i; _BitScanForward(&i, n); return i; } // 下位ビットに並ぶ 0 の個数
ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; } // 最大公約数
#define dump(x) cerr << "[DEBUG] " << endl << x << endl; // デバッグ出力用
#define dumpel(v) cerr << "[DEBUG]" << endl; for (const auto& x : v) {cout << x << endl;}
// 提出用（GCC）
#else
#define popcount (int)__builtin_popcount
#define popcountll (int)__builtin_popcountll
#define ctz __builtin_ctz
#define gcd __gcd
#define dump(x) 
#define dumpel(v) 
#endif

#endif // 無意味．折りたたむのが目的．


// AtCoder 専用
#include <atcoder/all>
using namespace atcoder;

// mint で使いたい法によってここを切り替える
using mint = modint1000000007;
//using mint = modint998244353;
//using mint = modint; // modint::set_mod(10000); // mint の法の指定

istream& operator>> (istream& is, mint& x) { ll tmp; is >> tmp; x = tmp; return is; } // mint の入力用
ostream& operator<< (ostream& os, const mint& x) { os << x.val(); return os; } // mint の出力用



//【コスト付きグラフの辺】
struct Edge {
	int to; // 行き先の頂点番号
	ll cost; // 辺のコスト

	// 出力
	friend ostream& operator<<(ostream& os, const Edge& e) {
		os << '(' << e.to << ',' << e.cost << ')';
		return os;
	}
};

//【グリッド → グラフ（改変）】O(H * W)
/*
* H 行 W 列のグリッドから nb_type 近傍を連結としたグラフを構築する．
* 壁マスは wall，空きマスはその他とする．
* i 行目の j 列目にあるマス (i, j) はグラフの g[i * W + j] に対応する．
*/
int u, d, r, l;
ll p;
void grid_to_graph(vector<vector<char>>& c, vector<vector<Edge>>& g) {
	int h = sz(c);
	int w = sz(c[0]);

	g = vector<vector<Edge>>(h * w);
	vector<ll> ruld = { r, u, l, d };

	rep(i, h) {
		rep(j, w) {
			// 空きマスでなかったら辺は追加しない．
			if (c[i][j] == '#') {
				continue;
			}

			// 今考えている近傍それぞれについて
			rep(k, 4) {
				// 近傍のマスの座標
				int ni = i + dy4[k];
				int nj = j + dx4[k];

				// 範囲外だったり空きマスでなかったら辺は追加しない．
				if (ni < 0 || ni >= h || nj < 0 || nj >= w || c[ni][nj] == '#') {
					continue;
				}

				// 近傍に空きマスがあったら辺を追加する．
				if (c[ni][nj] == '.') {
					g[i * w + j].push_back({ ni * w + nj, ruld[k] });
				}
				else {
					g[i * w + j].push_back({ ni * w + nj, ruld[k] + p });
				}
			}
		}
	}
}

//【ダイクストラ法】O(|E| log|V|)
/*
* 非負のコスト付きグラフ g に対し，始点を start として
* ダイクストラ法を用いて最小コスト経路問題を解き，
* start から各頂点 i への最小コストを cost[i] に格納する．
*/
void dijkstra(vector<vector<Edge>>& g, int start, vector<ll>& cost) {
	int n = (int)g.size();
	cost = vector<ll>(n, INFL); // スタートからの最小コストを保持するテーブル

	// 組 (スタートからのコスト, 頂点番号) を入れる優先度付きキューを用意する．
	// スタートからのコストがより小さいものを優先的に取り出す．
	priority_queue<pair<ll, int>, vector<pair<ll, int>>, greater<pair<ll, int>>> que;
	que.push(make_pair(0, start));

	while (!que.empty()) {
		auto c = que.top().first;
		auto v = que.top().second;
		que.pop();

		// もし既に最小コストが求まっているなら無視
		if (c >= cost[v]) {
			continue;
		}

		// 最小コストの決定
		// 優先度付きキューでコストの小さい順に取り出しているので最小の保証がある．
		cost[v] = c;

		// そこから移動できるノードについての情報をキューに追加する．
		for (auto e : g[v]) {
			que.push(make_pair(c + e.cost, e.to));
		}
	}
}

int main() {
	cout << fixed << setprecision(15); // 小数点以下の桁数の指定

	int h, w, xs, ys, xt, yt;
	ll k;
	cin >> h >> w >> u >> d >> r >> l >> k >> p >> xs >> ys >> xt >> yt;
	xs--;
	ys--;
	xt--;
	yt--;

	vector<vector<char>> c(h, vector<char>(w));
	cin >> c;

	vector<vector<Edge>> g;
	grid_to_graph(c, g);
	dumpel(g);

	vector<ll> cost;
	dijkstra(g, xs * w + ys, cost);

	Yes(cost[xt * w + yt] <= k);
}

yukicoder

結果

テストケース

ソースコード