제출 #927169

#제출 시각아이디문제언어결과실행 시간메모리
927169Gromp15Dango Maker (JOI18_dango_maker)C++17
13 / 100
1 ms500 KiB
#include <bits/stdc++.h>
#define ll long long
#define ar array
#define db double
#define all(x) x.begin(), x.end()
#define sz(x) (int)x.size()
using namespace std;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
#define rint(l, r) uniform_int_distribution<int>(l, r)(rng)
template<typename T> bool ckmin(T &a, const T &b) { return a > b ? a = b, 1 : 0; }
template<typename T> bool ckmax(T &a, const T &b) { return a < b ? a = b, 1 : 0; }
void test_case() {
	int n, m;
	cin >> n >> m;
	vector<string> a(n);
	for (auto &x : a) cin >> x;
	int ans = 0;
	vector<vector<bool>> vis(n, vector<bool>(m));
	const int dx[][2] = {{1, 2}, {-1, 1}, {-1, -2}};
	const string match = "RGW";
	auto in = [&](int x, int y) {
		return x>=0 && y>=0 && x<n && y<m;
	};
	auto check = [&](int x, int y, queue<ar<int, 2>> &v) {
		int idx = a[x][y] == 'R' ? 0 : a[x][y] == 'G' ? 1 : 2;
		bool ok = 1;
		for (int d = 0; d < 2; d++) {
			int nx = x + dx[idx][d];
			if (!in(nx, y) || a[nx][y] != match[idx + dx[idx][d]]) {
				ok = 0;
				break;
			}
		}
		if (ok) {
			if (!vis[x][y]) v.push({x, y}), vis[x][y] = 1;
			for (int d = 0; d < 2; d++) { 
				int nx = x + dx[idx][d];
				if (!vis[nx][y]) {
					vis[nx][y] = 1;
					v.push({nx, y});
				}
			}
		}
		bool ok2 = 1;
		for (int d = 0; d < 2; d++) {
			int ny = y + dx[idx][d];
			if (!in(x, ny) || a[x][ny] != match[idx + dx[idx][d]]) {
				ok2 = 0;
				break;
			}
		}
		if (ok2) {
			if (!vis[x][y]) v.push({x, y}), vis[x][y] = 1;
			for (int d = 0; d < 2; d++) { 
				int ny = y + dx[idx][d];
				if (!vis[x][ny]) {
					v.push({x, ny});
					vis[x][ny] = 1;
				}
			}
		}
	};
	vector<vector<int>> time(n, vector<int>(m));
	int timer = 1;
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < m; j++) if (!vis[i][j]) {
			queue<ar<int, 2>> q;
			check(i, j, q);
			vector<ar<int, 2>> cur;
			while (q.size()) {
				auto [x, y] = q.front();
				cur.push_back(q.front());
				q.pop();
				check(x, y, q);
			}
			auto check_right = [&](int x, int y) -> bool {
				if (y + 2 >= m) return 0;
				for (int d = 0; d < 3; d++) {
					if (a[x][y+d] != match[d] || time[x][y+d] == timer) {
						return 0;
					}
				}
				return 1;
			};
			auto check_down = [&](int x, int y) -> bool {
				if (x + 2 >= n) return 0;
				for (int d = 0; d < 3; d++) {
					if (a[x+d][y] != match[d] || time[x+d][y] == timer) {
						return 0;
					}
				}
				return 1;
			};
			if (cur.size()) {
				// choose horizontal / vertical
				sort(all(cur));
				int cnt1 = 0, cnt2 = 0;
				for (auto [x, y] : cur) {
					if (check_right(x, y)) {
						time[x][y] = time[x][y+1] = time[x][y+2] = timer;
						cnt1++;
					}
				}
				timer++;
				for (auto [x, y] : cur) {
					if (check_down(x, y)) {
						time[x][y] = time[x+1][y] = time[x+2][y] = timer;
						cnt2++;
					}
				}
				timer++;
				ans += max(cnt1, cnt2);
			}
		}
	}
	cout << ans << '\n';



}
int main() {
    cin.tie(0)->sync_with_stdio(0);
    int t = 1;
//    cin >> t;
    while (t--) test_case();
}

#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...