/*
HDOJ 1102
MST基本应用
*/
#include <iostream>
using namespace std;
#define INF 2000
int graph[103][103];
int lowcast[103];
int find[103];
int N,sum;
void Prim(int v)
{
int i,j,min,u;
for(i=1;i<=N;i++)
{
find[i]=0;
lowcast[i]=graph[v][i];
}
find[v]=1;
for(i=1;i<=N;i++)
{
min=INF;
for(j=1;j<=N;j++)
{
if(!find[j] && lowcast[j]<min)
{
min=lowcast[j];
u=j;
}
}
if(min == INF)
return;
find[u]=1;
sum += min;
for(j=1;j<=N;j++)
{
if(!find[j] && graph[u][j]<lowcast[j])
lowcast[j]=graph[u][j];
}
}
}
int main()
{
int i,j,m,a,b;
while(scanf("%d",&N) != EOF)
{
for(i=1;i<=N;i++)
for(j=1;j<=N;j++)
cin>>graph[i][j];
scanf("%d",&m);
for(i=1;i<=m;i++)
{
scanf("%d%d",&a,&b);
graph[a][b]=graph[b][a]=0; //已经建好的就直接设为0加上也不影响结果
}
sum=0;
Prim(1);
printf("%d\n",sum);
}
return 0;
}
HDOJ 1875 畅通工程再续
/*
HDOJ 1875 畅通工程
MST,输入是用矩阵的,所以用Prim来做比较方便
*/
#include <math.h>
#include <iostream>
#include <iomanip>
using namespace std;
#define INF 9999999.9
struct land
{
int x;
int y;
}Land[101];
double graph[101][101];
int find[101];
double lowcast[101];
int C,count;
double sum;
void Prim(int v)
{
int i,j,u;
double min;
for(i=0;i<C;i++)
{
find[i]=0;
lowcast[i]=graph[v][i];
}
find[v]=1;
for(i=1;i<C;i++)
{
min=INF;
for(j=0;j<C;j++)
{
if(!find[j] && lowcast[j]<min)
{
min=lowcast[j];
u=j;
}
}
if(min == INF)
return ;
find[u]=1;
sum += min;
count++;
for(j=0;j<C;j++)
{
if(!find[j] && graph[u][j]<lowcast[j])
lowcast[j]=graph[u][j];
}
}
}
int main()
{
int nCase,i,j,temp_1;
double temp_2;
cin>>nCase;
while(nCase--)
{
cin>>C;
for(i=0;i<C;i++)
{
cin>>Land[i].x>>Land[i].y;
}
for(i=0;i<C;i++)
{
graph[i][i]=0;
for(j=i+1;j<C;j++)
{
temp_1=(Land[i].x-Land[j].x)*(Land[i].x-Land[j].x)
+(Land[i].y-Land[j].y)*(Land[i].y-Land[j].y);
temp_2=sqrt(double(temp_1));
if(temp_2<10 || temp_2>1000)
graph[i][j]=graph[j][i]=INF;
else
graph[i][j]=graph[j][i]=temp_2;
}
}
sum=0.0;
count=0;
Prim(0);
if(count == C-1)
{
sum=sum*100;
cout<<setiosflags(ios::fixed);
cout<<setprecision(1)<<sum<<endl;
}
else
cout<<"oh!"<<endl;
}
return 0;
}