結果

問題 No.2290 UnUnion Find
ユーザー HIcoder
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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;
                    
                }

            }
            */

        }
    }
}
0