2020 Multi-University Training Contest 3---- HDU--6795、Little W and Contest（并查集）

题目链接

题面：

题意：

在ACM比赛中，三个人组成一个团队，其中这三个人中至少要有两个能力值为2，且这三个人中任意两个人不能熟悉。
熟悉关系有传递性，如果a与b相互熟悉，b与c相互熟悉，那么a与c相互熟悉。
现在给定 n 个人和他们的能力值 1或者2，每次会让两个不相互熟悉的人相互熟悉（操作n-1次之后，所有人都会相互熟悉），问每次选定两个不相互熟悉的人相互熟悉后，现在要组成一个团队，有多少种不同的组合方式。

题解：
考虑并查集维护每个相互熟悉的集合，和集合中能力值为2，能力值为1的人数。

假设现在要合并x，y两个集合。那么减少的贡献的组队方案就是：
（1）x集合选一个2，y集合选一个2，其他集合选一个 1 或者一个 2 .
（2）x集合选一个2，y集合选一个1，其他集合选一个2
（3）x集合选一个1，y集合选一个2，其他集合选一个2

代码：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<string>
#include<queue>
#include<bitset>
#include<map>
#include<unordered_map>
#include<set>
#include<list>
#define ui unsigned int
#define ll long long
#define llu unsigned ll
#define ld long double
#define pr make_pair
#define pb push_back
#define lc (cnt<<1)
#define rc (cnt<<1|1)
//#define len(x)  (t[(x)].r-t[(x)].l+1)
#define tmid ((l+r)>>1)
#define max(x,y) (x)>(y)?(x):(y)
#define min(x,y) (x)>(y)?(y):(x)
using namespace std;

const int inf=0x3f3f3f3f;
const ll lnf=0x3f3f3f3f3f3f3f3f;
const double dnf=1e18;
const int mod=1e9+7;
const double eps=1e-1;
const double pi=acos(-1.0);
const int hp=13331;
const int maxn=100100;
const int maxp=1100;
const int maxm=4000100;
const int up=1000;

ll f[maxn],si[maxn][3],cnt[3];

int fi(int x)
{
    if(f[x]!=x)
        f[x]=fi(f[x]);
    return f[x];
}

int main(void)
{
    int tt;
    scanf("%d",&tt);
    while(tt--)
    {
        int n,x,y;
        scanf("%d",&n);

        cnt[1]=cnt[2]=0;
        for(int i=1;i<=n;i++)
        {
            f[i]=i;
            si[i][1]=si[i][2]=0;
        }
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&x);
            cnt[x]++;
            si[i][x]++;
        }
        ll ans=(cnt[2]*(cnt[2]-1)*(cnt[2]-2)/6%mod+cnt[2]*(cnt[2]-1)/2%mod*cnt[1]%mod)%mod;
        printf("%lld\n",ans);
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&x,&y);
            x=fi(x),y=fi(y);
            ans=(ans-si[x][2]*si[y][2]*(cnt[2]+cnt[1]-si[x][2]-si[x][1]-si[y][2]-si[y][1])%mod+mod)%mod;
            ans=(ans-si[x][2]*si[y][1]*(cnt[2]-si[x][2]-si[y][2])%mod+mod)%mod;
            ans=(ans-si[x][1]*si[y][2]*(cnt[2]-si[x][2]-si[y][2])%mod+mod)%mod;
            f[x]=y;
            si[y][1]+=si[x][1];
            si[y][2]+=si[x][2];
            printf("%lld\n",ans);
        }
    }
    return 0;
}