SW练习_P3366_最小生成树_p3366 java-优快云博客

本文链接：https://blog.youkuaiyun.com/mush_me/article/details/106910698

本文深入探讨了Kruskal算法在解决最小生成树问题中的应用，详细解释了算法的实现过程，包括初始化并查集、边的优先级队列处理以及关键的并查集操作：查找、合并和判断连接性。通过具体实例展示了如何使用Kruskal算法找到加权图的最小生成树，并在不能形成连通图时输出特定信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这是一个模板题

唯一的要求就是，不能连通的话，让输出orz

package info.frady.luogu;


import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.PriorityQueue;
import java.util.StringTokenizer;

public class P3366 {
    static int[] set;
    static int N;
    public static void main(String[] args) throws Exception{
        BufferedReader reader=new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st=new StringTokenizer(reader.readLine());
        N=Integer.parseInt(st.nextToken());// N 个结点和 M 条无向边
        set=new int[N+1];
        int M=Integer.parseInt(st.nextToken());//M 条无向边
        PriorityQueue<Edge3366> pq=new PriorityQueue<Edge3366>();
        init();
        for (int i = 0; i <M ; i++) {
            st=new StringTokenizer(reader.readLine());
            int start=Integer.parseInt(st.nextToken());
            int end=Integer.parseInt(st.nextToken());
            int dis=Integer.parseInt(st.nextToken());
            pq.add(new Edge3366(start,end,dis));
        }
        int sum=0;
        int cout=0;
        while(!pq.isEmpty()){
            Edge3366 e=pq.poll();
            if(!connected(e.start,e.end)){
                union(e.start,e.end);
                sum=sum+e.dis;
                cout++;
            }
        }
        if(cout+1<N){
            System.out.println("orz");
        }else{
            System.out.println(sum);
        }
            reader.close();
        }

        public static void init(){
            for (int i = 0; i <set.length; i++) {
                set[i]=i;
            }
        }
        public static void union(int a,int b){
            int A=find(a);
            int B=find(b);
            if(A!=B){
             set[A]=B;
            }

        }
        public static int find(int a){
            if(set[a]==a){
                return set[a];
            }
            return set[a]=find(set[a]);

        }
        public static boolean connected(int a,int b){
            int A=find(a);
            int B=find(b);
            return A==B;
        }
}
class Edge3366 implements Comparable<Edge3366>{
    int start;
    int end;
    int dis;
    public Edge3366(int start, int end, int dis) {
        this.start = start;
        this.end = end;
        this.dis = dis;
    }

    @Override
    public int compareTo(Edge3366 o) {
        return this.dis-o.dis;
    }
}