java蓝桥杯练习 国王的烦恼
资源限制
时间限制:1.0s 内存限制:256.0MB
问题描述
C国由n个小岛组成,为了方便小岛之间联络,C国在小岛间建立了m座大桥,每座大桥连接两座小岛。两个小岛间可能存在多座桥连接。然而,由于海水冲刷,有一些大桥面临着不能使用的危险。
如果两个小岛间的所有大桥都不能使用,则这两座小岛就不能直接到达了。然而,只要这两座小岛的居民能通过其他的桥或者其他的小岛互相到达,他们就会安然无事。但是,如果前一天两个小岛之间还有方法可以到达,后一天却不能到达了,居民们就会一起抗议。
现在C国的国王已经知道了每座桥能使用的天数,超过这个天数就不能使用了。现在他想知道居民们会有多少天进行抗议。
输入格式
输入的第一行包含两个整数n, m,分别表示小岛的个数和桥的数量。
接下来m行,每行三个整数a, b, t,分别表示该座桥连接a号和b号两个小岛,能使用t天。小岛的编号从1开始递增。
输出格式
输出一个整数,表示居民们会抗议的天数。
样例输入
4 4
1 2 2
1 3 2
2 3 1
3 4 3
样例输出
2
样例说明
第一天后2和3之间的桥不能使用,不影响。
第二天后1和2之间,以及1和3之间的桥不能使用,居民们会抗议。
第三天后3和4之间的桥不能使用,居民们会抗议。
数据规模和约定
对于30%的数据,1<=n<=20,1<=m<=100;
对于50%的数据,1<=n<=500,1<=m<=10000;
对于100%的数据,1<=n<=10000,1<=m<=100000,1<=a, b<=n, 1<=t<=100000。
//java code
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();
int m=sc.nextInt();
int a,b,t;
//本题使用并查集的思想
/*
并查集,在一些有N个元素的集合应用问题中,我们通常是在开始时让每个元素构成一个单元素的集合,然后按一定顺序将属于同一组的元素所在的集合合并,其间要反复查找一个元素在哪个集合中。
主要操作:
初始化(把每个点所在集合初始化为其自身),通常来说,这个步骤在每次使用该数据结构时只需要执行一次,无论何种实现方式,时间复杂度均为O(N)。
查找(查找元素所在的集合,即根节点)
合并(将两个元素所在的集合合并为一个集合)
*/
ArrayList<Bridge> list=new ArrayList<Bridge>();
for (int i = 0; i < m; i++) {
a=sc.nextInt();
b=sc.nextInt();
t=sc.nextInt();
Bridge br= new Bridge(a,b,t);
list.add(br);
}
//将桥按天数进行降序排列
Collections.sort(list, new Comparator<Bridge>() {
@Override
public int compare(Bridge o1, Bridge o2) {
return o2.t-o1.t;//降序
}
});
UnionFind uf=new UnionFind(n);//初始化
int pt=-1,kt=0;//pt判断前后天数相同的标志,kt抗议天数
for (int i = 0; i < m; i++) {
if(!uf.connect(list.get(i).a,list.get(i).b)){
uf.union(list.get(i).a,list.get(i).b);
if(list.get(i).t!=pt){
kt++;
pt=list.get(i).t;
}
}
}
System.out.println(kt);
sc.close();
}
static class Bridge{
int a,b,t;
public Bridge(int a,int b, int t){
this.a=a;
this.b=b;
this.t=t;
}
}
static class UnionFind{
private int id[];
private int size[];
private int count;
public UnionFind(int N){
count=N;
id=new int[N+1];
size=new int[N+1];
for (int i = 1; i < N+1; i++) {
id[i]=i;
size[i]=1;
}
}
public int getCount() {
return count;
}
public boolean connect(int a, int b){
return find(a)==find(b);
}
public int find(int a){
if(a!=id[a])
id[a]=find(id[a]);
return id[a];
}
public void union(int a,int b){
int aroot=find(a);
int broot=find(b);
if (aroot==broot) return;
if(size[aroot]<size[broot]){
id[aroot]=broot;
size[broot]+=size[aroot];
}
else {
id[broot]=aroot;
size[aroot]+=size[broot];
}
count--;
}
}
}