每日一题 - 231130 - F - Colored Ball

• F - Colored Ball

---

• TAG - $set、启发式合并$
• 时间复杂度 - $O(N\ast logN)$

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

---

实现细节

• 启发式合并（将元素少的集合 合并到 元素多的集合）

---

参考示意图

• 无

---

参考链接

• C++ STL set insert()方法详解
• 算法进阶课——7.1 启发式合并、Manacher算法
• #ifdef #endif #ifndef 定义+代码详解

---

作者 | 乐意奥AI