結果
問題 | No.2290 UnUnion Find |
ユーザー |
|
提出日時 | 2023-05-05 21:34:09 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 422 ms / 2,000 ms |
コード長 | 2,519 bytes |
コンパイル時間 | 1,978 ms |
コンパイル使用メモリ | 208,900 KB |
最終ジャッジ日時 | 2025-02-12 17:38:32 |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 46 |
ソースコード
#include<bits/stdc++.h>using namespace std;//#pragma GCC optimize("Ofast")#define rep(i,n) for(ll i=0;i<n;i++)#define repl(i,l,r) for(ll i=(l);i<(r);i++)#define per(i,n) for(ll i=(n)-1;i>=0;i--)#define perl(i,r,l) for(ll i=r-1;i>=l;i--)#define fi first#define se second#define pb push_back#define ins insert#define pqueue(x) priority_queue<x,vector<x>,greater<x>>#define all(x) (x).begin(),(x).end()#define CST(x) cout<<fixed<<setprecision(x)#define rev(x) reverse(x);using ll=long long;using vl=vector<ll>;using vvl=vector<vector<ll>>;using pl=pair<ll,ll>;using vpl=vector<pl>;using vvpl=vector<vpl>;const ll MOD=1000000007;const ll MOD9=998244353;const int inf=1e9+10;const ll INF=4e18;const ll dy[8]={1,0,-1,0,1,1,-1,-1};const ll dx[8]={0,1,0,-1,1,-1,1,-1};template <typename T> inline bool chmax(T &a, T b) {return ((a < b) ? (a = b, true) : (false));}template <typename T> inline bool chmin(T &a, T b) {return ((a > b) ? (a = b, true) : (false));}struct UnionFind {vector<int> par;vector<int> edge;UnionFind(int n) : par(n, -1),edge(n, 0) {}int root(int x) {if (par[x] < 0) return x;else return par[x] = root(par[x]);}bool same(int x, int y) {return root(x) == root(y);}bool merge(int x, int y) {x = root(x); y = root(y);if (x == y) {edge[x]++;return false;}if (par[x] > par[y]) swap(x, y);par[x] += par[y];par[y] = x;edge[x] += edge[y]+1;return true;}int size(int x) {return -par[root(x)];}};int main(){ll n,q;cin >> n >> q;UnionFind uf(n);vector<vl> com(n);rep(i,n)com[i].emplace_back(i);vl siz(n,1);set<ll> st;rep(i,n)st.insert(i);rep(i,q){ll t;cin >> t;if(t==1){ll u,v;cin >> u >> v;u--;v--;if(uf.same(u,v))continue;ll f=uf.root(u)^uf.root(v);uf.merge(u,v);ll k=uf.root(u);f^=k;if(com[k].size()<com[f].size()){swap(com[k],com[f]);swap(siz[k],siz[f]);}for(auto p:com[f])com[k].emplace_back(p);siz[k]=com[k].size();siz[f]=0;com[f].clear();st.erase(f);}else{ll v;cin >> v;v--;v=uf.root(v);auto itr=st.begin();if(*itr==v)itr++;if(itr==st.end())cout << -1 << endl;else cout << com[*itr].front()+1 << endl;/*for(auto p:com[v]){cout << p <<" ";}cout << endl;*/}}}