108. 冗余连接
思路:并查集。每添加一条边,就判断这两个节点是否在统一集合。如果是,说明它是冗余的最后一条边,立即返回;否则把这两个节点添加到一个集合中。
只有 1 条冗余边。
拓展:树是一种无向无环图,n 个节点只需要 n - 1 条边就能完全联通。
import java.util.Scanner;
public class Main {
static int n = 1005;
static int[] father = new int[n];
static {
init();
}
static void init() {
for (int i = 0; i < n; i++) {
father[i] = i;
}
}
static int find(int a) {
return father[a] == a ? a : (father[a] = find(father[a]));
}
static void join(int a, int b) {
if (father[a] != father[b]) {
a = find(a);
b = find(b);
father[a] = b;
}
}
static boolean isSame(int a, int b) {
a = find(a);
b = find(b);
return a == b;
}
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
while (n-- > 0) {
int s = sc.nextInt();
int t = sc.nextInt();
if (isSame(s, t)) {
System.out.println(s + " " + t);
return;
} else {
join(s, t);
}
}
}
}
109. 冗余连接II
分为两大类情况:
- 存在入度为 2 的节点——判断删哪条边能成为一颗有向树。
- 可以任意删除一条(优先删除后添加的)(情况 1)
- 必须删除特定的一条(情况 2)
- 入度全部为 1,退化成和 108. 冗余连接 一样(情况 3)

自己写的:
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
public class Main {
static int n = 1005;
static int[] father = new int[n];
static {
init();
}
static void init() {
for (int i = 0; i < n; i++) {
father[i] = i;
}
}
static int find(int a) {
return father[a] == a ? a : (father[a] = find(father[a]));
}
static void join(int a, int b) {
if (father[a] != father[b]) {
a = find(a);
b = find(b);
father[a] = b;
}
}
static boolean isSame(int a, int b) {
a = find(a);
b = find(b);
return a == b;
}
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
List<int[]> edge = new ArrayList<>();
int[] inDegree = new int[n + 1]; // 统计入度为 2 的节点
for (int i = 0; i < n; i++) {
int s = sc.nextInt();
int t = sc.nextInt();
edge.add(new int[] {s, t});
inDegree[t]++;
}
int size = 0; // 统计入度为 2 的节点个数
int v = 0;
for (int i = 1; i < n + 1; i++) {
if (inDegree[i] == 2) {
size++;
v = i;
break;
}
}
if (size > 0) {
List<int[]> deleteEdge = new ArrayList<>(); // 哪两条边出度为 2
List<Integer> index = new ArrayList<>(); // 入度为 2 的边在 edge 中的序号是多少
for (int i = 0; i < n; i++) {
if (edge.get(i)[1] == v) {
deleteEdge.add(edge.get(i));
index.add(i);
}
}
if (isTreeAfterDelete(edge, index.get(1))) { // 优先删后添加的边
System.out.println(deleteEdge.get(1)[0] + " " + deleteEdge.get(1)[1]);
} else {
System.out.println(deleteEdge.get(0)[0] + " " + deleteEdge.get(0)[1]);
}
} else {
for (int i = 0; i < n; i++) {
int s = edge.get(i)[0];
int t = edge.get(i)[1];
if (isSame(s, t)) {
System.out.println(s + " " + t);
return;
} else {
join(s, t);
}
}
}
}
private static boolean isTreeAfterDelete(List<int[]> edge, int index) {
for (int i = 0; i < edge.size(); i++) {
if (i == index) {
continue;
}
int s = edge.get(i)[0];
int t = edge.get(i)[1];
if (isSame(s, t)) {
return false;
} else {
join(s, t);
}
}
return true;
}
}
优化后的代码:
把
int size = 0; // 统计入度为 2 的节点个数
int v = 0;
for (int i = 1; i < n + 1; i++) {
if (inDegree[i] == 2) {
size++;
v = i;
break;
}
}
优化成:
List<Integer> indexOfDeleteEdge = new ArrayList<>(); // 入度为 2 的边在 edge 中的序号是多少
for (int i = 0; i < n; i++) {
if (inDegree[edge.get(i)[1]] == 2) { // 最精妙的地方,不是直接判断 inDegree[i] == 2,那样只会添加 1 次,表示哪个节点入度为 2
indexOfDeleteEdge.add(i); // 应该会添加 2 次。保证了后添加的边在后边。
}
}
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
public class Main {
static int n = 1005;
static int[] father = new int[n];
static {
init();
}
static void init() {
for (int i = 0; i < n; i++) {
father[i] = i;
}
}
static int find(int a) {
return father[a] == a ? a : (father[a] = find(father[a]));
}
static void join(int a, int b) {
if (father[a] != father[b]) {
a = find(a);
b = find(b);
father[a] = b;
}
}
static boolean isSame(int a, int b) {
a = find(a);
b = find(b);
return a == b;
}
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
List<int[]> edge = new ArrayList<>();
int[] inDegree = new int[n + 1]; // 统计入度为 2 的节点
for (int i = 0; i < n; i++) {
int s = sc.nextInt();
int t = sc.nextInt();
edge.add(new int[] {s, t});
inDegree[t]++;
}
List<Integer> indexOfDeleteEdge = new ArrayList<>(); // 入度为 2 的边在 edge 中的序号是多少
for (int i = 0; i < n; i++) {
if (inDegree[edge.get(i)[1]] == 2) { // 最精妙的地方,不是直接判断 inDegree[i] == 2,那样只会添加 1 次,表示哪个节点入度为 2
indexOfDeleteEdge.add(i); // 应该会添加 2 次。保证了后添加的边在后边。
}
}
if (indexOfDeleteEdge.size() > 0) {
int index1 = indexOfDeleteEdge.get(0); // 待删除的边在 edge 中的索引
int index2 = indexOfDeleteEdge.get(1);
int s1 = edge.get(index1)[0];
int t1 = edge.get(index1)[1];
int s2 = edge.get(index2)[0];
int t2 = edge.get(index2)[1];
if (isTreeAfterDelete(edge, index2)) { // 优先删后添加的边
System.out.println(s2 + " " + t2);
} else {
System.out.println(s1 + " " + t1);
}
} else { // 不删任何边,把造成不能成为有向树的冗余边找出来
for (int i = 0; i < n; i++) {
int s = edge.get(i)[0];
int t = edge.get(i)[1];
if (isSame(s, t)) {
System.out.println(s + " " + t);
return;
} else {
join(s, t);
}
}
}
}
// 删了这条边之后,能否成为有向树
private static boolean isTreeAfterDelete(List<int[]> edge, int index) {
for (int i = 0; i < edge.size(); i++) {
if (i == index) {
continue;
}
int s = edge.get(i)[0];
int t = edge.get(i)[1];
if (isSame(s, t)) {
return false;
} else {
join(s, t);
}
}
return true;
}
}
254

被折叠的 条评论
为什么被折叠?



