POJ1751 PRIM基础题（模板）

那个mst函数是我自己写的。有点超时。

学习了一下别人的prim模板。

以后要改进了。

#include<iostream>

#include<cmath>
using namespace std;
int n,m;
const int N=755;
const int M=1005;
const int inf=99999999;
struct 
{
int x,y;
}town[N];
double mat[N][N];
bool visit[N];
void mst()
{
visit[1]=true;
for(int count=1;count<n;count++)
{
int min=inf;
int mina=0,minb=0;
for(int i=1;i<=n;i++)
{
if(visit[i])
{
for(int j=1;j<=n;j++)
{
if(!visit[j])
{
if(min>mat[i][j])
min=mat[i][j],mina=i,minb=j;
}
}
}
}
visit[minb]=true;
if(min!=0)
printf("%d %d\n",mina,minb);
}
}
double key[N];
int pre[N];
void prim()
{
int k;
for(int i=1;i<=n;i++)
key[i]=mat[1][i],pre[i]=1;
visit[1]=true;
for(int i=1;i<n;i++)
{
int min=inf;
for(int j=1;j<=n;j++)
{
if(!visit[j]&&min>key[j])
{
min=key[j];
k=j;
}
}
visit[k]=true;
if(mat[k][pre[k]]!=0)
printf("%d %d\n",k,pre[k]);
for(int j=1;j<=n;j++)
{
if(!visit[j]&&key[j]>mat[k][j])
pre[j]=k,key[j]=mat[k][j];
}
}
}
int main()
{
//freopen("C:\\Users\\wuyanyisb\\Desktop\\1.txt","r",stdin);
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&town[i].x,&town[i].y);
}
scanf("%d",&m);
int a,b;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
mat[i][j]=inf;
}
for(int i=1;i<=m;i++)
{
scanf("%d%d",&a,&b);
mat[a][b]=0;
mat[b][a]=0;
}
for(int i=1;i<=n;i++)
for(int j=i+1;j<=n;j++)
{
if(mat[i][j]!=0)
{
int tmp=(town[i].x-town[j].x)*(town[i].x-town[j].x)+(town[i].y-town[j].y)*(town[i].y-town[j].y);
mat[i][j]=tmp;
mat[j][i]=mat[i][j];
}
}
//mst();
prim();
return 0;
}