結果

問題 No.2277 Honest or Dishonest ?
ユーザー Carpenters-Cat
提出日時 2023-04-22 02:08:04
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 113 ms / 2,000 ms
コード長 1,615 bytes
コンパイル時間 2,387 ms
コンパイル使用メモリ 205,456 KB
最終ジャッジ日時 2025-02-12 12:50:02
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 50
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using Vi = std::vector<int> ;
using VVi = vector<Vi>;
int par[200020], siz[200020];
void init(int N) {
    iota(par, par + N, 0);
    fill(siz, siz + N, 1);
}
int root(int a) {
    return (a == par[a] ? a : root(par[a]));
}
void merge(int a, int b) {
    a = root(a); b = root(b);
    if (a == b) {
        return;
    }
    if (siz[a] < siz[b]) {
        swap(a, b);
    }
    par[b] = a;
    siz[a] += siz[b];
}
int main() {
    int N, Q;
    cin >> N >> Q;
    init(N);
    vector<pair<int, int>> X;
    for (int i = 0; i < Q; i ++) {
        int a, b, c;
        cin >> a >> b >> c;
        if (c) {
            X.emplace_back(--a, --b);
        } else {
            merge(--a, --b);
        }
    }
    VVi gr(N);
    for (auto [a, b] : X) {
        a = root(a); b = root(b);
        gr[a].push_back(b);
        gr[b].push_back(a);
    }
    using ll = long long;
    ll const m = 998244353;
    ll ans = 1;
    int did[200002];
    fill(did, did + N, -1);
    for (int i = 0; i < N; i ++) {
        if (i != root(i) || did[i] != -1) {
            continue;
        }
        int ok = 2;
        queue<int> que;
        did[i] = 0;
        que.push(i);
        while (!que.empty()) {
            int u = que.front();
            que.pop();
            for (auto v : gr[u]) {
                if (did[v] == -1) {
                    did[v] = 1 - did[u];
                    que.push(v);
                } else if (did[v] == did[u]) {
                    ok = 0;
                }
            }
        }
        ans = (ans * ok) % m;
    }
    cout << ans << endl;
}
0