</pre><pre code_snippet_id="428667" snippet_file_name="blog_20140717_1_2418943" name="code" class="cpp"></pre><pre code_snippet_id="428667" snippet_file_name="blog_20140717_1_2418943" name="code" class="cpp">#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f
int map[110][110];
int lowcost[110];
bool visit[110];
int nodenum,sum;
void prim()
{
int temp,k;
sum=0;
memset(visit,false,sizeof(visit));
visit[1]=true;
for(int i=1;i<=nodenum;i++)
lowcost[i]=map[1][i];
for(int i=1;i<=nodenum;i++)
{
temp=INF;
for(int j=1;j<=nodenum;j++)
if(!visit[j]&&temp>lowcost[j])
temp=lowcost[k=j];
if(temp==INF)
break;
visit[k]=true;
sum=sum+temp;
for(int j=1;j<=nodenum;j++)
if(!visit[j]&&lowcost[j]>map[k][j])
lowcost[j]=map[k][j];
}
// printf("%d\n",sum);
}
int main()
{
int a,b,p,cost,edgenum;
while(scanf("%d",&nodenum)!=EOF)
{
// for(i=1;i<=nodenum;i++)
// for(j=1;j<=nodenum;j++)
// map[i][j]=INF;
if(nodenum==0)
break;
memset(map,INF,sizeof(map));
edgenum=nodenum*(nodenum-1)/2;
for(int i=0;i<edgenum;i++)
{
scanf("%d%d%d%d",&a,&b,&cost,&p);
if(p==1)
map[a][b]=map[b][a]=0;
if(p==0)
{
if(cost<map[a][b])
map[a][b]=map[b][a]=cost;
}
}
prim();
printf("%d\n",sum);
// for(int i=1;i<nodenum;i++)
// for(int j=i+1;j<=nodenum;j++)
// printf("%d \n",map[i][j]);
}
return 0;
}
hdu1879继续畅通工程
最新推荐文章于 2021-05-11 20:46:10 发布