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;}
}