Java实现 蓝桥杯VIP 算法训练 连通分块（并查集）

试题 算法训练 连通分块

资源限制

时间限制：200ms 内存限制：8.0MB

问题描述

　　连通分块

输入格式

　　输入的第一行包含两个整数n, m

　　n代表图中的点的个数，m代表边的个数

　　接下来m行，每行2个正整数，表示图中连通的两点。

输出格式

　　输出1行，与1连通的点的集合，并按升序排列输出。

样例输入

6 3

1 2

2 3

3 4

样例输出

1 2 3 4

数据规模和约定

　　n<=10000,m<=100000

package 蓝桥杯官网;

import java.util.Scanner;

public class 连通分块 {

    //https://blog.csdn.net/a1439775520/article/details/90746562  欢迎进入讨论群
    public static int[] f;

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        f = new int[n + 1];
        int x, y;
        for (int i = 1; i <= n; i++) {
            f[i] = i;
        }
        for (int i = 0; i < m; i++) {
            x = sc.nextInt();
            y = sc.nextInt();
            setinfo(x, y);
        }
        for (int i = 1; i <= n; i++) {
            if (f[i] == 1) System.out.print(i + " ");
        }

    }

    public static void setinfo(int a, int b) {
        int fa = find(a);
        int fb = find(b);
        if (fa > fb) {
            f[fa] = fb;
        } else {
            f[fb] = fa;
        }
    }

    public static int find(int num) {
        if (num != f[num]) {
            int temp = find(f[num]);
            f[num] = temp;
        }
        return f[num];
    }
}