結果

問題 No.3695 同室と別室
コンテスト
ユーザー kwm_t
提出日時 2026-09-09 21:36:39
言語 C++23
(gcc 15.3.0 + boost 1.92.0 + ACL)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 41 ms / 2,000 ms
+ 729µs
コード長 3,509 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 4,060 ms
コンパイル使用メモリ 384,964 KB
実行使用メモリ 26,240 KB
最終ジャッジ日時 2026-09-09 21:36:45
合計ジャッジ時間 5,999 ms
ジャッジサーバーID
(参考情報)
judge1_0 / judge3_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 13
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

#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; }
#ifndef KWM_T_GRAPH_BIPARTITE_COLORING_HPP
#define KWM_T_GRAPH_BIPARTITE_COLORING_HPP

#include <vector>

namespace kwm_t::graph::bipartite {
/**
 * @brief 二部グラフ判定 + 2色塗り
 *
 * 典型用途:
 *   - グラフが二部グラフか判定
 *   - 頂点を 0/1 に色分け
 *   - 連結成分数の取得
 *
 * 計算量:
 *   - O(N + M)
 *
 * @param n 頂点数
 *
 * 制約 / 注意:
 *   - 無向グラフを想定
 *
 * 使用例:
 *   vector graph(n, vector<int>());
 *   BipartiteColoring bc(n,graph);
 *   auto col = bc.get_color; // 0 or 1
 *
 * verified:
 *   - https://atcoder.jp/contests/awc0071/submissions/75933094
 */
class BipartiteColoring {
public:
	explicit BipartiteColoring(int n, const std::vector<std::vector<int>>& graph)
		: n(n), graph(graph), color(n, -1), components(0) {
		execute();
	}

	bool is_bipartite() const {
		return bipartite;
	}

	int get_components() const {
		return components;
	}

	int get_color(int i) const {
		return color[i];
	}

	const std::vector<int>& get_color() const {
		return color;
	}


private:
	int n;
	bool bipartite;
	std::vector<std::vector<int>> graph;
	std::vector<int> color;
	int components;

	bool dfs(int v) {
		bool ok = true;
		for (int nv : graph[v]) {
			if (color[nv] != -1) {
				if (color[nv] == color[v]) ok = false;
			}
			else {
				color[nv] = 1 - color[v];
				if (!dfs(nv)) ok = false;
			}
		}
		return ok;
	}

	void execute() {
		bipartite = true;
		components = 0;

		for (int i = 0; i < n; ++i) {
			if (color[i] != -1) continue;
			color[i] = 0;
			components++;
			if (!dfs(i)) bipartite = false;
		}
	}
};

} // namespace kwm_t::graph::bipartite

#endif // KWM_T_GRAPH_BIPARTITE_COLORING_HPP
int main() {
	std::ios::sync_with_stdio(false);
	std::cin.tie(nullptr);

	int n, q; cin >> n >> q;
	vector g(n, vector<P>());
	dsu uf(n);
	rep(i, q) {
		int t; cin >> t;
		int a, b; cin >> a >> b;
		a--, b--;
		uf.merge(a, b);
		g[a].emplace_back(b, t);
		g[b].emplace_back(a, t);
	}
	mint ans = 1;
	vector<int>col(n, -1);
	rep(i, n)if (uf.leader(i) == i) {
		auto dfs = [&](auto&& dfs, int v)->bool {
			for (auto [nv, t] : g[v]) {
				if (t == 0) {
					if (col[nv] != -1) {
						if (col[nv] != col[v])return false;
					}
					else {
						col[nv] = col[v];
						auto get = dfs(dfs, nv);
						if (!get)return false;
					}
				}
				else {
					if (col[nv] != -1) {
						if (col[nv] == col[v])return false;
					}
					else {
						col[nv] = 1 - col[v];
						auto get = dfs(dfs, nv);
						if (!get)return false;
					}
				}
			}
			return true;
		};
		col[i] = 0;
		auto get = dfs(dfs, i);
		if (get)ans *= 2;
		else ans *= 0;
	}
	cout << ans.val() << endl;
	return 0;
}
0