結果

問題 No.2277 Honest or Dishonest ?
ユーザー achapi
提出日時 2023-04-21 23:20:06
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,267 bytes
コンパイル時間 4,327 ms
コンパイル使用メモリ 262,032 KB
最終ジャッジ日時 2025-02-12 12:33:19
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 19 WA * 31
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include <atcoder/all>
using mint = atcoder::modint998244353;
using namespace std;

int main() {
	int n, m;
	cin >> n >> m;
	atcoder::dsu uf(n);
	vector<vector<pair<int, int>>> g(n);
	for (int i = 0; i < m; i++){
		int a, b, c;
		cin >> a >> b >> c;
		a--;
		b--;
		g[a].push_back({b, c});
		uf.merge(a, b);
	}
	mint ans = 1;
	vector<int> f(n, -1);
	vector<bool> f2(n, true);
	for (auto s : uf.groups()){
		int ok = 2;
		for (int i : s){
			if (f[i] != -1){
				continue;
			}
			queue<int> q;
			q.push(i);
			f[i] = 0;
			while (!q.empty()){
				int v = q.front();
				q.pop();
				for (auto p : g[v]){
					if (f[p.first] == -1){
						f[p.first] = (f[v] ^ p.second);
						q.push(p.first);
					}else{
						if (f[p.first] != (f[v] ^ p.second)){
							ok = 0;
						}
					}
				}
			}
			if (ok == 0){
				ok = 2;
				f[i] = 1;
				q.push(i);
				while (!q.empty()){
					int v = q.front();
					q.pop();
					for (auto p : g[v]){
						if (f2[p.first]){
							f[p.first] = (f[v] ^ p.second);
							q.push(p.first);
						}else{
							if (f[p.first] != (f[v] ^ p.second)){
								ok = 0;
							}
						}
						f2[p.first] = false;
					}
				}
			}
			if (ok == 0){
				break;
			}
		}
		ans *= ok;
	}
	cout << ans.val() << endl;
}
0