【Prim】-POJ-2421-构造公路

题目链接：http://poj.org/problem?id=2421

题目描述：最裸的最短路题目

解题思路：

明明是一个最裸的最短路题。周赛那天竟然一直WA。唉，后来自己试试，竟然又WA，拜托。。这是最裸体啊。。我到底怎么了？！然后找张纸画了画，发现prim函数里写得太混乱了，错了好几个地方，看来是我没理解透彻，唉~说来例会那天也挂黑板上了。。下次再也不溜号了。。= =  555555 T_T

AC代码：

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

const int INF=9999999;
int cost[1015][1015],dis[1015];
bool vis[1015];
int N;

int prim()
{
int from,to,ans=0,p,i;
from=1;
vis[1]=1;
for(to=1;to<=N;to++)
dis[to]=cost[from][to];
while(1)
{
int mindis=INF;
for(to=1;to<=N;to++)
{
if(dis[to]<mindis&&!vis[to])
{
mindis=dis[to];
p=to;
}
}
if(mindis==INF)break;
vis[p]=1;
ans+=mindis;
from=p;
for(to=1;to<=N;to++)
{
dis[to]=min(dis[to],cost[from][to]);
}
}
return ans;
}

int main()
{
freopen("2421_input.txt","r",stdin);
int ans,i,j,num;
while(cin>>N)
{
for(i=1;i<=N;i++)
dis[i]=INF;
memset(vis,0,sizeof(vis));
for(i=1;i<=N;i++)
for(j=1;j<=N;j++)
{
cin>>cost[i][j];
}
cin>>num;
while(num--)
{
cin>>i>>j;
cost[i][j]=cost[j][i]=0;
}
ans=prim();
cout<<ans<<endl;
}
return 0;
}


AC截图：