結果
問題 | No.2277 Honest or Dishonest ? |
ユーザー |
![]() |
提出日時 | 2023-04-21 21:39:07 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 85 ms / 2,000 ms |
コード長 | 2,801 bytes |
コンパイル時間 | 3,844 ms |
コンパイル使用メモリ | 255,900 KB |
最終ジャッジ日時 | 2025-02-12 11:13:25 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 50 |
コンパイルメッセージ
main.cpp: In function ‘int potentialized_dsu<S>::merge(int, int, S) [with S = int]’: main.cpp:27:27: warning: ‘t’ may be used uninitialized [-Wmaybe-uninitialized] 27 | v = t - v; | ~~^~~~~~~ main.cpp:26:27: note: ‘t’ was declared here 26 | S t; | ^
ソースコード
#include <stdio.h> #include <bits/stdc++.h> #include <atcoder/all> using namespace atcoder; using mint = modint998244353; using namespace std; #define rep(i,n) for (int i = 0; i < (n); ++i) #define Inf32 1000000001 #define Inf64 1000000000000000001 template <class S> struct potentialized_dsu { public: potentialized_dsu() : _n(0) {} potentialized_dsu(int n) : _n(n), parent_or_size(n, -1), d(n) {} int merge(int a, int b, S v) { assert(0 <= a && a < _n); assert(0 <= b && b < _n); int x = leader(a), y = leader(b); if (x == y) return x; if (-parent_or_size[x] < -parent_or_size[y]){ std::swap(x, y); std::swap(a, b); S t; v = t - v; } v += diff(x,a); v -= diff(y,b); d[y] += d[x] + v; parent_or_size[x] += parent_or_size[y]; parent_or_size[y] = x; return x; } bool same(int a, int b) { assert(0 <= a && a < _n); assert(0 <= b && b < _n); return leader(a) == leader(b); } int leader(int a) { assert(0 <= a && a < _n); if (parent_or_size[a] < 0) return a; int l = leader(parent_or_size[a]); if(parent_or_size[a] != l){ d[a] += d[parent_or_size[a]]; parent_or_size[a] = l; } return parent_or_size[a]; } int size(int a) { assert(0 <= a && a < _n); return -parent_or_size[leader(a)]; } S diff(int a,int b){ assert(0 <= a && a < _n); assert(0 <= b && b < _n); leader(a); leader(b); return d[b] - d[a]; } std::vector<std::vector<int>> groups() { std::vector<int> leader_buf(_n), group_size(_n); for (int i = 0; i < _n; i++) { leader_buf[i] = leader(i); group_size[leader_buf[i]]++; } std::vector<std::vector<int>> result(_n); for (int i = 0; i < _n; i++) { result[i].reserve(group_size[i]); } for (int i = 0; i < _n; i++) { result[leader_buf[i]].push_back(i); } result.erase( std::remove_if(result.begin(), result.end(), [&](const std::vector<int>& v) { return v.empty(); }), result.end()); return result; } private: int _n; // root node: -1 * component size // otherwise: parent std::vector<int> parent_or_size; std::vector<S> d; }; int main(){ int n,q; cin>>n>>q; vector<int> a(q),b(q),c(q); potentialized_dsu<int> D(n); rep(i,q){ cin>>a[i]>>b[i]>>c[i]; a[i]--,b[i]--; if(c[i]==0)D.merge(a[i],b[i],0); } rep(i,q){ if(c[i]==1){ if(D.same(a[i],b[i])){ if(D.diff(a[i],b[i])%2==0){ cout<<0<<endl; return 0; } } D.merge(a[i],b[i],1); } } mint ans = mint(2).pow(D.groups().size()); cout<<ans.val()<<endl; return 0; }