HDU 1102 最小生成树 prim

#include<cstdio>
#include<iostream>
const int inf=1002;
using namespace std;

int a[101][101],dis[101],visit[101];

int prim(int N)//1为原点
{
int i,sum,min,mark,j,k;
for(i=2;i<=N;i++)
dis[i]=a[1][i];

dis[1]=0;
visit[1]=1;

sum=0;
for(i=1;i<N;i++)
{
min=inf;
for(k=1;k<=N;k++)
{
if(visit[k]==0&&min>dis[k])
{
mark=k;
min=dis[k];

}

}
sum+=min;

visit[mark]=1;
dis[mark]=0;

for(j=1;j<=N;j++)
{
if(dis[j]>dis[mark]+a[mark][j]&&visit[j]==0)
dis[j]=dis[mark]+a[mark][j];
}

}

return sum;

}

int main()
{
int N,i,j,Q;
while(scanf("%d",&N)!=EOF)
{
for(i=1;i<=N;i++)
{
a[i][i]=0;
visit[i]=0;
for(j=1;j<=N;j++)
{
scanf("%d",&a[i][j]);

}
}

scanf("%d",&Q);
while(Q--)
{
scanf("%d%d",&i,&j);
a[i][j]=0;
a[j][i]=0;
}
int sum=prim(N);
printf("%d\n", sum);
}
return 0;

}