Problem E
Time Limit : 2000/1000ms (Java/Other) Memory Limit : 32768/32768K (Java/Other)
Total Submission(s) : 178 Accepted Submission(s) : 55
3 1 2 1 0 1 3 2 0 2 3 4 0 3 1 2 1 0 1 3 2 0 2 3 4 1 3 1 2 1 0 1 3 2 1 2 3 4 1 0
3 1 0
解题思路:想先将已经建好的公路建成一棵树,再kruskal;
ac代码
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define inf 0x3f3f3f3f
int map[100][100];
int s[100],vis[100];
int n,m;
int prim()
{
int i,j,t,p,min,cnt,minpos;
int ans=0;
cnt=0;
vis[1]=1;
s[cnt++]=1;
while(cnt<n)
{
t=cnt;
min=inf;
for(i=0;i<t;i++)
{
p=s[i];
for(j=1;j<=n;j++)
{
if(!vis[j]&&map[p][j]<min)
{
min=map[p][j];
minpos=j;
}
}
}
ans+=min;
s[cnt++]=minpos;
vis[minpos]=1;
}
return ans;
}
int main()
{
int i,sum;
while(~scanf("%d",&n)&&n)
{
memset(vis,0,sizeof(vis));
memset(map,inf,sizeof(map));
int b,c,d,sta;
m=n*(n-1)/2;
for(i=0;i<m;i++)
{
scanf("%d%d%d%d",&b,&c,&d,&sta);
if(sta==0)
map[b][c]=map[c][b]=d;
else
map[b][c]=map[c][b]=0;
}
sum=prim();
printf("%d\n",sum);
}
return 0;
}