POJ 1258 Agri-Net（最小生成树，模板题）

用的是prim算法。

我用vector数组,每次求最小的dis时，不需要遍历所有的点，只需要遍历之前加入到vector数组中的点(即dis[v]!=INF的点)。
但其实时间也差不多，和遍历所有的点的方法都是16ms。。。

#include <iostream>
#include <algorithm>
#include <string.h>
#include <stdio.h>
#include <string>
#include <queue>
using namespace std;

const int INF=0x3f3f3f3f;
int n,cost;
int ans;
int w[101][101];
int dis[101]; //生成树外的点与生成树相连的最短边长
int pre[101]; //pre[v]为v的前驱节点，用来输出进入最小生成树的边。该题用不到
int vis[101]; //标记点是否在生成树中
vector<int> son[101];

void init() {
memset(pre,0,sizeof(pre));
memset(dis,INF,sizeof(dis));
memset(vis,0,sizeof(vis));
}

int solve() {
vector<int> node;
int s=1,counts=0,ans=0,tmp,k;
dis[s]=0;
pre[s]=s;

node.push_back(s);
while(1) {
tmp=INF;

for(int i=0; i<node.size(); i++) {
int v=node[i];
if(!vis[v]&& dis[v]<tmp) {
tmp=dis[v];
k=v;   //k即为在没有进入最小生成树的点中到树的距离（dis[k]）最小的点。
}
}
/*
//直接遍历所有的点
for(int i=1;i<=n;i++){
if(!vis[i]){
if(dis[i]<tmp){
tmp=dis[i];
k=i;
}
}
}
*/
if(tmp==INF)
break;
ans+=tmp;
vis[k]=1;

for(int i=0; i<son[k].size(); i++) {
int v=son[k][i];
if(!vis[v] && w[k][v]<dis[v]) {
dis[v]=w[k][v];
pre[v]=k;
node.push_back(v);
}
}
}
return ans;

}
int main() {
while(scanf("%d",&n)!=EOF) {

for(int i=0; i<101; i++) {
son[i].clear();
}
for(int i=1; i<=n; i++) {
for(int j=1; j<=i; j++)
scanf("%d",&cost);
for(int j=i+1; j<=n; j++) {
scanf("%d",&cost);
w[i][j]=w[j][i]=cost;
son[i].push_back(j);
son[j].push_back(i);
}
}
ans=0;
init();
ans=solve();

printf("%d\n",ans);
}
return 0;
}