Java实现版最小生成树Prim算法

	import java.util.*;
	public class Leetcode {
		 public static void main(String args[]) {
			 Scanner scan=new Scanner(System.in);
			 int n=scan.nextInt();
			 int m=scan.nextInt();
			 int cost[][]=new int[n+1][n+1];
			 for(int i=1;i<n+1;i++) {
				 for(int j=1;j<n+1;j++)
					 cost[i][j]=Integer.MAX_VALUE;
			 }
			 for(int i=0;i<m;i++) {
				 int x=scan.nextInt();
				 int y=scan.nextInt();
				 int z=scan.nextInt();
				 if(z<cost[x][y]) {
				 cost[x][y]=z;
				 cost[y][x]=z;
				 }
				 
			 }
			 long dp[]=new long [n+1];
			 boolean isused[]=new boolean[n+1];
			 Arrays.fill(dp, Integer.MAX_VALUE);
			 dp[1]=0;
			 int res=0;
			 while(true) {
				 int v=-1;
				 for(int i=1;i<n+1;i++) {
					 if(!isused[i]&&(v==-1||dp[i]<dp[v])) v=i;
				 }
				 if(v==-1) break;
				 isused[v]=true;
				 res+=dp[v];
				 for(int i=1;i<n+1;i++) {
					 if(dp[i]>cost[v][i]) {
						 dp[i]=cost[v][i];
					 }
				 }
			 }
				 
			
		
				 System.out.print(res);
			 
			 
	}
	
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值