牛客OI周赛11-普及组 C: Colorful ( 二进制枚举 + 最小生成树 )

题目链接

解题报告：
二进制枚举颜色，考虑所枚举颜色的集合对答案连通性的贡献。

#define first f
#define second s
#define ll long long
#define mp make_pair
#define pb push_back
#define pf push_front
#define lb lower_bound
#define ub upper_bound
#define pii pair<int,int>
#include <bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
const int MOD=1e9+7;
const int maxn=3e5+5;
const double eps=1e-9;
const int inf=0x3f3f3f3f;
const double PI=acos(-1.0);
const double e=2.718281828459;

vector<pii>p[20];
int f[maxn];
int ma[20];
unordered_map<int,bool>MP[105];

int getfind(int x)
{
    return x==f[x]?x:f[x]=getfind(f[x]);
}
int main()
{
    int T,n,m,s,u,v,w,pp,cnt,ant;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d%d",&n,&m,&s);
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&u,&v,&w);
            if(!MP[u].count(v)) p[w].pb({u,v}),MP[u][v]=true,MP[v][u]=true;
        }
        ant=inf;
        for(int i=1,j;i<(1<<s);i++){
            j=i;pp=cnt=0;
            while(j){ma[++pp]=j&1;j/=2;cnt+=ma[pp];}
            for(int i=1;i<=n;i++) f[i]=i;
            int ans=1;
            for(int k=1;k<=pp;k++){
                if(!ma[k]) continue;
                for(int q=0;q<p[k].size();q++){
                    u=getfind(p[k][q].f);
                    v=getfind(p[k][q].s);
                    if(u!=v){
                        f[u]=v;ans++;
                    }
                }
            }
            if(ans==n) ant=min(ant,cnt);
        }
        printf("%d\n",ant);
        for(int i=1;i<=s;i++) p[i].clear();
        for(int i=1;i<=n;i++) MP[i].clear();
    }
    return 0;
}