Prim思想:
选择任意一个起点,这个点就是贪心策略刚开始的一个集合,这个集合就是一个整张图最小生成树的一部分(子集),然后通过贪心策略不断扩大这个子集。直到这个最小生成树集合包含了所有的点。
首先需要一个表,保存所有点到最开始最小生成树点集合的距离,不能到达就是INF;
然后,选择一个距离集合最近的一个点,加入集合(最小生成树点的集合),加入集合的点距离集合距离为0,下次不再计算;
通过这个新加入的点,更新集合到其它未加入的点的距离。
最后经过N-1步,就得到了最小生成树。
package com.yxs.graph1;
import java.io.BufferedInputStream;
import java.util.Arrays;
import java.util.Scanner;
import java.util.Stack;
public class Prim {
static class Edge{
int to,w,next;
{
to = 0;
w = 0;
next = 0;
}
}
static final int MAXN = 105;
static int[] head = new int[MAXN];
static Edge[] edge = new Edge[MAXN];
static int len = 1;
static void add(int from, int to, int w) {
edge[len] = new Edge();
edge[len].to = to;
edge[len].w = w;
edge[len].next = head[from];
head[from] = len++;
}
static int[] dist = new int[MAXN];//dist[i]保存i到最小生成树集合最近的距离
static int[] path = new int[MAXN];//path[i]保存顶点i接入最小生成树中的结点位置
static final int INF = 0x3f3f3f3f;
static {
Arrays.fill(dist, INF);
}
static void prim(int start, int n) {
for(int i = head[start];i != 0;i = edge[i].next) {
dist[edge[i].next] = edge[i].w;
}
Arrays.fill(path, 1, n+1, start);
dist[start] = 0;
int ind, mind, sum = 0;
//下面所说的集合内,表示已经生成好的最小生成树集合,最开始是一个点。
//集合外所代表的是,还没有加入最小生成树集合的点
for(int i = 1;i < n;i++) {//最小生成树只需要执行n-1步
ind = 0;
mind = INF;
for(int j = 1;j <= n;j++) {//找一个当前最小生成树集合(最开始是一个点)能够到达的最近的点
if(dist[j] != 0 && dist[j] < mind) {
mind = dist[j];
ind = j;
}
}
if(mind == INF) break;//没有找到
sum += mind;//计入最小生成树权值之和
dist[ind] = 0;//加入最小生成树集合,集合扩大,下次不再使用该点
for(int j = head[ind];j != 0;j = edge[j].next) {
//通过找到的这个点,对其它在集合外的点进行松弛,看能不能通过找到的这个点缩短到集合外其它点的距离
if(dist[edge[j].to] != 0 && edge[j].w < dist[edge[j].to]) {
dist[edge[j].to] = edge[j].w;
path[edge[j].to] = ind;
}
}
}
System.out.println("最小生成树权值为:" + sum);
}
public static void main(String[] args) {
Scanner in = new Scanner(new BufferedInputStream(System.in));
int n = in.nextInt();
int e = in.nextInt();
for(int i = 1;i <= e;i++) {
int from = in.nextInt();
int to = in.nextInt();
int w = in.nextInt();
add(from, to, w);
}
prim(1, n);
}
}
/*
6 8
1 6 100
1 5 30
1 3 10
2 3 5
3 4 50
4 6 10
5 4 20
5 6 60
*/