結果
問題 | No.2290 UnUnion Find |
ユーザー |
![]() |
提出日時 | 2023-05-09 12:49:59 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 388 ms / 2,000 ms |
コード長 | 978 bytes |
コンパイル時間 | 2,026 ms |
コンパイル使用メモリ | 175,892 KB |
実行使用メモリ | 13,440 KB |
最終ジャッジ日時 | 2024-11-26 00:02:44 |
合計ジャッジ時間 | 21,422 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 46 |
ソースコード
#include <bits/stdc++.h> using namespace std; struct UnionFind { vector< int > data; set<int> st; UnionFind(int sz) { data.assign(sz, -1); for( int i = 0; i < sz; i++ ) st.insert( i ); } bool unite(int x, int y) { x = find(x), y = find(y); if(x == y) return (false); if(data[x] > data[y]) swap(x, y); data[x] += data[y]; data[y] = x; st.erase( y ); return (true); } int find(int k) { if(data[k] < 0) return (k); return (data[k] = find(data[k])); } int size(int k) { return (-data[find(k)]); } int ex( int r ) { auto it = st.begin(); while( *it == r ) it++; return *it; } }; int main() { int N, Q; cin >> N >> Q; UnionFind uf( N ); for( int i = 0; i < Q; i++ ) { int t, u, v; cin >> t >> u; u--; if( t == 1 ) { cin >> v; v--; uf.unite( u, v ); } else { int ans = -1; int r = uf.find( u ); if( uf.size( r ) < N ) ans = uf.ex( r ) + 1; cout << ans << endl; } } }