最小生成树prim算法hdu1879

本暑假集训第一次用prim写最小生成树;各种调试:最后发现多加一个边orz,只有我这种人才这样可怜;码代码这么挫

这也算是最小生成树prim的模板吧:

#include<algorithm>
#include<cstdio>
#include<string.h>
#include<queue>
#include<cmath>
using namespace std;
const int N=102;
int dist[N];
int map[102][102];
const int INF=999999;
bool vis[102];
int prim(int n,int m)
{
	memset(vis,false,sizeof(vis));
	for(int i=0;i<=n;i++)
	{  
		for(int j=0;j<=n;j++)
		{
			if(i!=j)
		 map[i][j]=INF;
			else
				map[i][j]=0;
		}
		
	}
	int u,v,val,ok=0;
	
	for(int i=0;i<m;i++)
	{
	  scanf("%d%d%d%d",&u,&v,&val,&ok);
	   if(ok)
		  val=0;
	  // if(map[u][v]>val)
	  map[v][u]=map[u][v]=val;
	 }
	for(int i=1;i<=n;i++)
	{
	 dist[i]=map[1][i];
	}
	dist[1]=INF;
	vis[1]=true;
	int Min,sum=0,p=-1;

	for(int i=1;i<n;i++)
	{
	 Min=INF;
	 for(int j=1;j<=n;j++)
	 {
	  if(!vis[j] && dist[j] < Min)
	  {
	   Min=dist[j];
	   p=j;
	  }
	 }
	  sum+=dist[p];
	 // printf("dist[%d]= %d\n",p,dist[p]);
	  vis[p]=true;
	  
	  for(int j=1;j<=n;j++)
	  {
	   if(!vis[j] && dist[j] > map[p][j])
	   {
	     dist[j]=map[p][j];
	   }
	  }
	}
	return sum;
}
int main()
{
	int n;
	while(~scanf("%d",&n)&&n)
	{
	 printf("%d\n",prim(n,n*(n-1)/2));
	}
  return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值