每日一题 - 231130 - F - Colored Ball

原创于 2023-11-30 18:00:47 发布 · 619 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#算法

2023 专栏收录该内容

29 篇文章

订阅专栏

本文介绍了如何在C++中使用STLset数据结构实现F-ColoredBall问题的启发式合并算法，通过将元素较少的集合合并到较多的集合，优化了时间复杂度至O(N*logN)。作者还讨论了相关算法和代码实现细节。

F - Colored Ball

TAG - $se t 、启发式合并$
时间复杂度 - $\ast logN)$

//
#include<bits/stdc++.h>
using namespace std;
// #define int long long 

const int N=2e5+6;
set<int> st[N];

void solve()
{
    int n,q;
    scanf("%d%d",&n,&q );
    for( int i=1,x;i<=n;i++ ) scanf("%d",&x ),st[i].insert(x);

    while( q-- )
    {
        int a,b;
        scanf("%d%d",&a,&b );

        if( st[a].size()>st[b].size() ) swap( st[a],st[b] );
        st[b].insert( st[a].begin(),st[a].end() );
        st[a].clear();

        printf("%d\n",st[b].size() );
    }
}

signed main()
{
    int t=1;
    // scanf("%d",&t );
    while( t-- ) solve();
    return 0;
}