結果
問題 |
No.2290 UnUnion Find
|
ユーザー |
|
提出日時 | 2023-07-07 10:40:32 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 342 ms / 2,000 ms |
コード長 | 3,500 bytes |
コンパイル時間 | 1,188 ms |
コンパイル使用メモリ | 108,368 KB |
最終ジャッジ日時 | 2025-02-15 06:34:15 |
ジャッジサーバーID (参考情報) |
judge5 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 46 |
ソースコード
#include<iostream> #include<set> #include<algorithm> #include<vector> #include<string> #include<set> #include<map> #include<numeric> #include<queue> #include<cmath> using namespace std; typedef long long ll; const ll INF=1LL<<60; typedef pair<int,int> P; typedef pair<int,P> PP; const ll MOD=998244353; const double PI=acos(-1); class UnionFind{ public: std::vector<int> par; std::vector<int> rank; std::vector<int> sz;//sz[i]で,頂点iが含まれているグループのサイズ std::vector<int> root; UnionFind(int size){ rank=std::vector<int>(size+1,0); par = std::vector<int>(size+1,0); std::iota(par.begin(),par.end(),0);//#include<numeric> sz = std::vector<int>(size+1,1); root=std::vector<int>(size+1); std::iota(root.begin(),root.end(),0); } ~UnionFind(){ std::vector<int>().swap(rank); std::vector<int>().swap(par); std::vector<int>().swap(sz); std::vector<int>().swap(root); } int find(int x){ if(par[x]==x)return x; else return par[x] = find(par[x]); } bool issame(int u,int v){ return find(u)==find(v); } void merge(int u,int v){ u = find(u); v = find(v); if(u==v) return; if(rank[u]<rank[v]) std::swap(u,v); //uの傘下へvが入る //rank[u]>=rank[v] par[v]=u; sz[u]+=sz[v]; if(rank[u]==rank[v])rank[u]++; } int size(int u){//頂点uが属すグループの大きさを表す. u=find(u); return sz[u]; } std::map<int,std::vector<int>> element(){ std::map<int,std::vector<int>> mp; for(int i=0;i<par.size();i++){ root[i]=find(i); mp[root[i]].push_back(i); } return mp; } }; int main(){ int N,Q; cin>>N>>Q; //vector<int> idmax(N),idmin(N); //iota(idmax.begin(),idmax.end(),0); //iota(idmin.begin(),idmin.end(),0); UnionFind uf(N); int l=0,r=N-1; //set<int> st; for(int q=0;q<Q;q++){ int t; cin>>t; if(t==1){ int u,v; cin>>u>>v; u--;v--; uf.merge(u,v); while(r!=uf.find(r))r--; while(l!=uf.find(l))l++; /* if(uf.issame(u,v))continue; int cmax=max(idmax[uf.find(u)],idmax[uf.find(v)]); int cmin=min(idmin[uf.find(u)],idmin[uf.find(v)]); if(idmax[uf.find(u)]<idmax[uf.find(v)]){ swap(u,v); } int vmax=idmax[uf.find(v)]; st.erase(vmax); int umax=idmax[uf.find(u)];//<idmax[uf.find(v)] uf.merge(u,v); */ }else{ int v; cin>>v; v--; if(l==r){ cout<<-1<<endl; }else if(uf.issame(l,v)){ cout<<r+1<<endl; }else{ cout<<l+1<<endl; } /* if(uf.size(v)==N){ cout<<-1<<endl; }else{ int rootv=uf.find(v); auto it=st.begin(); if(*it !=rootv){ cout<<*it<<endl; }else{ it++; cout<<*it<<endl; } } */ } } }