Leetcode1579-保证图可完全遍历

并查集

基于贪心的思想,我们应当优先删除仅Alice或者仅Bob通过的边,尽可能的保留下二者都可通行的公共边。因此,我们对所有的边排序,把公共边都放在前边,然后用并查集维护连通性。

先对Alice能在图上通过的边都遍历一遍,用并查集来维护点之间的连通性,如果已经连通,则说明该边可以删除,对该边打上标记。然后枚举所有点,看看并查集中有多少个连通块,如果大于一个,说明Alice无法到达整个图,返回-1。

同理,再对Bob进行如上操作。

最后统计一下所有被打上标记的边即可。

class Solution {
public:
    vector<int> p=vector<int>(100010);
    int findd(int x){
        return p[x]==x?x:p[x]=findd(p[x]);
    }
    void unionn(int x,int y){
        p[findd(x)]=findd(y);
    }
    int maxNumEdgesToRemove(int n, vector<vector<int>>& edges) {
        sort(edges.begin(),edges.end(),[](const auto &a,const auto &b){
            return a[0]>b[0];
        });
        int m=edges.size();
        vector<int> d(m,0);
        int ans=0;
        iota(p.begin(),p.end(),0);
        for(int i=0;i<m;i++){
            if(edges[i][0]==2) continue;
            int x=findd(edges[i][1]),y=findd(edges[i][2]);
            if(x==y) d[i]=1;
            else unionn(x,y);
        }
        int cnt=0;
        for(int i=1;i<=n;i++) if(p[i]==i) cnt++;
        if(cnt>1) return -1;
        cnt=0;
        iota(p.begin(),p.end(),0);
        for(int i=0;i<m;i++){
            if(edges[i][0]==1) continue;
            int x=findd(edges[i][1]),y=findd(edges[i][2]);
            if(x==y) d[i]=1;
            else unionn(x,y);
        }
        for(int i=1;i<=n;i++) if(p[i]==i) cnt++;
        if(cnt>1) return -1;
        for(int i=0;i<m;i++) if(d[i]) ans++;
        return ans;
    }
};

时间复杂度:O(n+m·α(n)),α(n)为并查集操作的复杂度

空间复杂度:O(n)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值