結果

問題 No.2884 Pieces on Squares
コンテスト
ユーザー hiro1729
提出日時 2024-09-01 16:36:06
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 2,060 bytes
コンパイル時間 3,249 ms
コンパイル使用メモリ 259,984 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-09-06 21:41:52
合計ジャッジ時間 5,616 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 5
other WA * 45
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
#include <atcoder/dsu>

int solve(int H, int W, std::vector<std::string> S) {
	atcoder::dsu uf(H + W);
	for (int i = 0; i < H; i++) {
		for (int j = 0; j < W; j++) {
			if (S[i][j] == '@') {
				uf.merge(i, H + j);
			}
		}
	}
	std::vector<std::vector<int>> g = uf.groups();
	int L = g.size();
	std::vector<int> c(L), d(L);
	for (int i = 0; i < L; i++) {
		for (int u: g[i]) {
			if (u < H) c[i]++;
			else d[i]++;
		}
	}
	std::vector<int> dp1(H + 1, -1), dp2(H + 1, 1e9);
	dp1[0] = 0;
	dp2[0] = 0;
	for (int i = 0; i < L; i++) {
		for (int j = H; j >= 0; j--) {
			if (dp1[j] != -1 && j + c[i] <= H) {
				dp1[j + c[i]] = std::max(dp1[j + c[i]], dp1[j] + d[i]);
				dp2[j + c[i]] = std::min(dp2[j + c[i]], dp2[j] + d[i]);
			}
		}
	}
	int ans = 0;
	for (int i = 0; i <= H; i++) {
		if (dp1[i] != -1) ans = std::max(ans, i * (W - dp1[i]) + (H - i) * dp1[i]);
		if (dp2[i] != 1e9) ans = std::max(ans, i * (W - dp2[i]) + (H - i) * dp2[i]);
	}
	return ans;
}

int naive(int H, int W, std::vector<std::string> S) {
	int ans = 0;
	for (int i = 0; i < (1 << H); i++) {
		for (int j = 0; j < (1 << W); j++) {
			bool ok = true;
			for (int k = 0; k < H; k++) {
				for (int l = 0; l < W; l++) {
					if (S[k][l] == '@' && ((i >> k) & 1) != ((j >> l) & 1)) ok = false;
				}
			}
			if (ok) {
				int c = __builtin_popcount(i), d = __builtin_popcount(j);
				ans = std::max(ans, c * (W - d) + d * (H - c));
			}
		}
	}
	return ans;
}

int main() {
	/*
	int H = 4, W = 4;
	std::random_device rd;
	std::mt19937 en(rd());
	std::vector<std::string> S(H);
	for (int i = 0; i < H; i++) {
		for (int j = 0; j < W; j++) {
			if (en() % 100 < 50) {
				S[i] += '@';
			} else {
				S[i] += '.';
			}
		}
	}
	int ans1 = solve(H, W, S), ans2 = naive(H, W, S);
	std::cout << ans1 << ' ' << ans2 << '\n';
	if (ans1 != ans2) {
		for (int i = 0; i < H; i++) {
			std::cout << S[i] << '\n';
		}
	}
	*/
	int H, W;
	std::cin >> H >> W;
	std::vector<std::string> S(H);
	for (int i = 0; i < H; i++) {
		std::cin >> S[i];
	}
	std::cout << solve(H, W, S) << '\n';
}
0