Codeforces Round #764 (Div. 3) G. MinOr Tree

G. MinOr Tree

题目链接

题意：

给定n个顶点和m条边（无向边），其中m>=n-1。
求：保留n-1条边，并且是图为连通图（不存在环），使得n-1条边的权值进行按位与（“|”）运算，使得权值和（所有边“按位与和”）最小。

思路与代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=2e5+5;
void Get_YES_NO(int flag){printf("%s\n",(flag?"YES":"NO"));}
inline int read()
{
    int s=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){s=s*10+ch-'0';ch=getchar();}
    return s*f;
}
ll x[MAXN],y[MAXN],w[MAXN];
int vis[MAXN];
//并查集模板
class UnionFind {
private:
    vector<int> f, sz;
    // 当前连通分量数目
    int setCount;
public:
    //初始化
    UnionFind(int n): f(n+1), sz(n+1) {
        for (int i = 1; i <= n; i++) {
            f[i] = i;
            sz[i] = 1;
        }
        setCount = n;
    }
    //找x的父节点，并压缩路径
    int find(int x) {
        if (f[x] == x) {
            return x;
        }
        int newf = find(f[x]);
        f[x] = newf;
        return f[x];
    }
    //合并x,y
    bool merge(int x, int y) {
        int fx = find(x), fy = find(y);
        if (fx == fy) {
            return false;
        }
        f[fx] = fy;
        sz[fy] += sz[fx];
        --setCount;
        return true;
    }
    //x所在树中的大小
    int size(int x) {
        return sz[find(x)];
    }
    //返回联通数
    int GetSetCount(){
        return setCount;
    }
};

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    int t;cin>>t;
    while(t--){
        ll n,m;cin>>n>>m;
        for(int i=1;i<=m;i++){
            vis[i]=0;
            cin>>x[i]>>y[i]>>w[i];
        }
        ll ans=0;
        for(ll k=29;k>=0;k--){
            UnionFind uf(n);
            for(int i=1;i<=m;i++){
                if(!vis[i]&&!((1<<k)&w[i]))uf.merge(x[i],y[i]);
            }
            if(uf.GetSetCount()==1){
                //如果枚举的二进制（记为res）的最高位可以不要，并且还能组合成连通图，则去掉这些与res相关并且没有标记的边
                for(int i=1;i<=m;i++){
                    if(!vis[i]&&((1<<k)&w[i]))vis[i]=1;
                }
            }else {
                //如果去掉res的最高位，就无法组合成连通图，则这个值必须要保留
                ans+=(1<<k);
            }
        }
        cout<<ans<<endl;
    }
    return 0;
}