杭电OJ1232畅通工程Java实现并查集

it2023-11-11  83

最近在读《算法第四版》学习了并查集算法,通过一道题来巩固自己的知识。也希望能帮助到刷这道题的人。 题目链接

题目描述

Problem Description

某省调查城镇交通状况,得到现有城镇道路统计表,表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通(但不一定有直接的道路相连,只要互相间接通过道路可达即可)。问最少还需要建设多少条道路? Input 测试输入包含若干测试用例。每个测试用例的第1行给出两个正整数,分别是城镇数目N ( < 1000 )和道路数目M;随后的M行对应M条道路,每行给出一对正整数,分别是该条道路直接连通的两个城镇的编号。为简单起见,城镇从1到N编号。 注意:两个城市之间可以有多条道路相通,也就是说 3 3 1 2 1 2 2 1 这种输入也是合法的 当N为0时,输入结束,该用例不被处理。 Output 对每个测试用例,在1行里输出最少还需要建设的道路数目。 Sample Input 4 2 1 3 4 3 3 3 1 2 1 3 2 3 5 2 1 2 3 5 999 0 0 Sample Output 1 0 2 998

代码

import java.util.Scanner; public class Main { static int[] id; static int[] sz; static int find(int a) { return a == id[a] ? a : find(id[a]); } static void union(int a, int b) { if (a == b) return; int x = find(a), y = find(b); if (sz[x] >= sz[y]) { id[y] = x; sz[x] += sz[y]; } else { id[x] = y; sz[y] += sz[x]; } } public static void main(String[] args) { Scanner sc = new Scanner(System.in); while (sc.hasNext()) { int N = sc.nextInt(); if (N == 0) break; int M = sc.nextInt(); id = new int[N + 1]; sz = new int[N + 1]; for (int i = 0; i < id.length; i++) { id[i] = i; sz[i] = 1; } for (int i = 0; i < M; i++) { int x = sc.nextInt(), y = sc.nextInt(); union(x, y); } int count=0; for (int i = 1; i <id.length; i++) { if(i==id[i])count++; } System.out.println(count-1); } } }
最新回复(0)