HDU 6109 百度之星2017 1002 数据分割(并查集+set)

题目链接

百度之星2017　1002

分析

个比赛的时候想到了并查集+set的做法，只是评测实在太慢了，也没仔细想set处理合并的复杂度，赛后发现原来直接暴力合并set就行了　(不能分析复杂度是多少)

AC code

//Problem : 6109 ( 数据分割 )     Judge Status : Accepted
//RunId : 21701687    Language : G++    Author : zouzhitao
//Code Render Status : Rendered By HDOJ G++ Code Render Version 0.01 Beta
#include<bits/stdc++.h>
#define pb push_back
#define mp make_pair
#define PI acos(-1)
#define fi first
#define se second
#define INF 0x3f3f3f3f
#define INF64 0x3f3f3f3f3f3f3f3f
#define random(a,b) ((a)+rand()%((b)-(a)+1))
#define ms(x,v) memset((x),(v),sizeof(x))
#define scint(x) scanf("%d",&x );
#define scf(x) scanf("%lf",&x );
#define eps 1e-10
#define dcmp(x) (fabs(x) < eps? 0:((x) <0?-1:1))
#define lc o<<1
#define rc o<<1|1
using namespace std;
typedef long long LL;
typedef long double DB;
typedef pair<int,int> Pair;
const int maxn = 1e5+10;
const int maxv = 1e6+10;

std::vector<int> ans;

set<int> x[maxn],st;
int p[maxn];
int FIND(int x){
    return p[x] == x? x :p[x] = FIND(p[x]);
}
int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    int n;cin>>n;
    for(int i=1 ; i<= maxn; ++i)p[i] = i;
    int last =0;
    for(int i=1 ; i<=n ; ++i){
        int a,b,c;cin>>a>>b>>c;
        int fa = FIND(a) , fb = FIND(b);
        st.insert(a);st.insert(b);
        if(c){
            if(x[fa].count(fb) || x[fb].count(fa)){
                ans.pb(i - last);
                last = i;
                for(auto e: st){
                    p[e] = e;
                    x[e].clear();
                }
                st.clear();
            }else if(fa != fb){
                //合并集合
                if(x[fa].size() > x[fb].size())swap(fa,fb);
                p[fa] = fb;
                for(auto e : x[fa]){
                    x[fb].insert(e);
                    x[e].erase(fa);
                    x[e].insert(fb);
                }
                x[fa].clear();
            }
        }else {
            x[a].insert(b);x[b].insert(a);
            if(fa  == fb){
                ans.pb(i - last);
                last = i;
                for(auto e : st){
                    p[e] = e;
                    x[e].clear();
                }
                st.clear();
            }
        }
    }
    std::cout << ans.size() << '\n';
    for(auto e : ans)std::cout << e << '\n';
    return 0;
}