代码随想录算法训练营第 54 天 | 108. 冗余连接、109. 冗余连接II

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

题目链接

leetcode 685

分为两大类情况:

  • 存在入度为 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;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值