蓝桥杯必练数据结构-并查集

148 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路

并查集

并查集是数据结构中常用的一种算法,通过给每个元素划分类型,计算出总共有多少个类型等问题。

如问题:

H - 畅通工程 HDU - 1232 某省调查城镇交通状况,得到现有城镇道路统计表,表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通(但不一定有直接的道路相连,只要互相间接通过道路可达即可)。问最少还需要建设多少条道路?

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

解决思路:

将能相通的城镇划分成一个集合,最后会形成n个集合,那n个集合就需要n-1条路来进行连通。

Coding

 import java.io.*;
 import java.util.*;
 ​
 public class Main{
     static BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
     static BufferedWriter out = new BufferedWriter(new OutputStreamWriter(System.out));
     static int[] a = new int[1005]; //假定最大能设置1005个城镇
     static int find(int k) {
         if(a[k] == k) return k;
         return a[k] = find(a[k]);
     }
     public static void main(String[] args) throws Exception{
         String[] str = null;
         while((str = in.readLine().split(" ")) != null){
             int n = Integer.parseInt(str[0]);
             if(n == 0) break;
             int ans = 0;
             int road = Integer.parseInt(str[1]);
             for(int i = 0; i < n; ++i) {
                 a[i] = i;
             }
             while((road--) != 0) {
                 str = in.readLine().split(" ");
                 int x = Integer.parseInt(str[0]);
                 int y = Integer.parseInt(str[1]);
                 a[find(x)] = find(y);
             }
             for(int i = 0; i < n; ++i) {
                 if(a[i] == i) 
                     ++ans;
             }
             out.write(ans-1 + "\n");
             out.flush();    
         }
     }
 }