华为OD题目： 计算至少需要多少个快递主站点

华为OD题目： 计算至少需要多少个快递主站点

时间限制: 1s 空间限制: 256MB 限定语言: 不限
题目描述:
快递业务范围有N个站点，A站点与B站点可以中转快递，则认为A-B站可达，如果A-B可达，B-C可达，则A-C可达。现在给N个站点编号0、1、…n-1，用s]表示-j是否可达，s]=1表示i-j可达，sD]=0表示i-j不可达。
现用二维数组给定N个站点的可达关系，请计算至少选择从几个主站点出发，才能可达所有站点(覆盖所有站点业
)
说明: sij与s小i取值相同

输入描述:
第一行输入为N，N表示站点个数。
之后N行表示站点之间的可达关系，第i行第i个数值表示编号为i和i之间是否可达
输出描述
输出站点个数，表示至少需要多少个主站点。
补充说明:
1<N<10000

示例1
输入:
4
1 1 1 1
1 1 1 0
1 1 1 0
1 0 0 1
输出:
1

说明:
选择0号站点作为主站点，0站点可达其他所有站点，所以至少选择1个站点作为主站才能覆盖所有站点业务

示例2

输入:
4
1 1 0 0
1 1 0 0
0 0 1 0
0 0 0 1
输出:
3
说明:
选择0号站点可以覆盖0、1站点，选择2号站点可以覆盖2号站点，选择3号站点可以覆盖3号站点，所以至少选择3个站点作为主站才能覆盖所有站点业务

输入:
4
1 1 0 1
1 1 0 0
0 0 1 0
1 0 0 1
输出:
3

解题思路：

• dfs 从0开始遍历站点，如果set里面没有，那个加入到set里面，然后dfs与当前站点所有联通的站点，加入到set里面

public class My {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String str = sc.nextLine();
        int n = Integer.parseInt(str);

        int[][] arr = new int[n][n];
        for (int i = 0; i < n; i++) {
            String line = sc.nextLine();
            String[] strings = line.split(" ");
            for (int j = 0; j < n; j++) {
                arr[i][j] = Integer.parseInt(strings[j]);
            }
        }

        int result = 0;
        //已经有连通的站点
        Set<Integer> set = new HashSet<>();
        for (int i = 0; i < n; i++) {
            if (set.contains(i)) {
                continue;
            }
            set.add(i);
            dfs(set, arr, i);
            result++;
        }
        System.out.println(result);

    }

    /**
     * 找到所有跟index相连的站点，放入set里面
     * @param set
     * @param arr
     * @param index
     */
    public static void dfs(Set<Integer> set,int[][] arr,  int index) {
        for (int i = index; i < arr.length; i++) {
            if (set.contains(i)) {
                continue;
            }
            if (arr[index][i] == 1) {
                set.add(i);
                dfs(set, arr, i);
            }
        }
    }


}