Alice 和 Bob 共有一个无向图,其中包含 n 个节点和 3 种类型的边:
类型 1:只能由 Alice 遍历。
类型 2:只能由 Bob 遍历。
类型 3:Alice 和 Bob 都可以遍历。
给你一个数组 edges ,其中 edges[i] = [typei, ui, vi] 表示节点 ui 和 vi 之间存在类型为 typei 的双向边。请你在保证图仍能够被 Alice和 Bob 完全遍历的前提下,找出可以删除的最大边数。如果从任何节点开始,Alice 和 Bob 都可以到达所有其他节点,则认为图是可以完全遍历的。
返回可以删除的最大边数,如果 Alice 和 Bob 无法完全遍历图,则返回 -1 。
示例 1:
输入:n = 4, edges = [[3,1,2],[3,2,3],[1,1,3],[1,2,4],[1,1,2],[2,3,4]]
输出:2
解释:如果删除 [1,1,2] 和 [1,1,3] 这两条边,Alice 和 Bob 仍然可以完全遍历这个图。再删除任何其他的边都无法保证图可以完全遍历。所以可以删除的最大边数是 2 。
代码
class Solution {int[] fa;public void init(){for(int i=0;i<fa.length;i++)fa[i]=i;}public int find(int x){if(x!=fa[x])fa[x]=find(fa[x]);return fa[x];}public void union(int x,int y){x=find(x);y=find(y);if(x==y) return;fa[x]=y;}public int maxNumEdgesToRemove(int n, int[][] edges) {int n1=n,n2=n,res=0;fa=new int[n+1];init();for(int[] c:edges)//先把公共边放进去{if(c[0]==3){if(find(c[1])==find(c[2])){res++;}else{union(c[1],c[2]);n1--;n2--;}}}int[] temp=fa.clone();for(int[] c:edges)//在遍历完公关边的基础上,遍历类型为1的所有边{if(c[0]==1){if(find(c[1])==find(c[2])){res++;}else{union(c[1],c[2]);n1--;}}}fa=temp;for(int[] c:edges)//在遍历完公关边的基础上,遍历类型为2的所有边{if(c[0]==2){if(find(c[1])==find(c[2])){res++;}else{union(c[1],c[2]);n2--;}}} if(n1!=1||n2!=1)//如果alice和bob其中一个不能遍历所以节点,则返回-1return -1;else return res;}
}