結果

問題 No.2731 Two Colors
ユーザー kwm_tkwm_t
提出日時 2024-04-20 13:52:52
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 247 ms / 3,000 ms
コード長 2,127 bytes
コンパイル時間 3,317 ms
コンパイル使用メモリ 258,940 KB
実行使用メモリ 11,500 KB
最終ジャッジ日時 2024-10-12 09:25:25
合計ジャッジ時間 7,825 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
//#include <atcoder/all>
using namespace std;
//using namespace atcoder;
//using mint = modint1000000007;
//const int mod = 1000000007;
//using mint = modint998244353;
//const int mod = 998244353;
//const int INF = 1e9;
//const long long LINF = 1e18;
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define rep2(i,l,r)for(int i=(l);i<(r);++i)
#define rrep(i, n) for (int i = (n) - 1; i >= 0; --i)
#define rrep2(i,l,r)for(int i=(r) - 1;i>=(l);--i)
#define all(x) (x).begin(),(x).end()
#define allR(x) (x).rbegin(),(x).rend()
#define P pair<int,int>
template<typename A, typename B> inline bool chmax(A & a, const B & b) { if (a < b) { a = b; return true; } return false; }
template<typename A, typename B> inline bool chmin(A & a, const B & b) { if (a > b) { a = b; return true; } return false; }
const int dx[4] = { 1,0,-1,0 };
const int dy[4] = { 0,-1,0,1 };
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int h, w; cin >> h >> w;
	vector a(h, vector<int>(w));
	rep(i, h)rep(j, w)cin >> a[i][j];
	vector col(h, vector<int>(w, -1));
	col[0][0] = 0;
	col[h - 1][w - 1] = 1;
	int cnt = 0;
	using T = pair<int, P>;
	vector< std::priority_queue<T, std::vector<T>, std::greater<T>>>pq(2);
	rep(i, 2) {
		rep(j, 4) {
			int x, y;
			if (0 == i) {
				x = 0 + dx[j];
				y = 0 + dy[j];
			}
			else {
				x = h - 1 + dx[j];
				y = w - 1 + dy[j];
			}
			if (x < 0 || x >= h)continue;
			if (y < 0 || y >= w)continue;
			pq[i].push({ a[x][y],{x,y} });
		}
	}
	while (true) {
		int t = cnt % 2;
		cnt++;
		while (true) {
			auto top = pq[t].top();
			if (-1 != col[top.second.first][top.second.second]) {
				pq[t].pop();
				continue;
			}
			break;
		}
		auto top = pq[t].top();
		pq[t].pop();
		col[top.second.first][top.second.second] = t;
		bool chk = false;
		rep(i, 4) {
			int x = top.second.first + dx[i];
			int y = top.second.second + dy[i];
			if (x < 0 || x >= h)continue;
			if (y < 0 || y >= w)continue;
			if (col[x][y] == 1 - t) {
				chk = true;
			}
			if (-1 != col[x][y])continue;
			pq[t].push({ a[x][y],{x,y} });
		}
		if (chk)break;
	}
	cout << cnt << endl;
	return 0;
}
0