usaco 3.1 Agri-Net

/*
ID: modengd1
PROG: agrinet
LANG: C++
*/
#include <iostream>
#include <stdio.h>
#include <memory.h>
#include <queue>
using namespace std;
int N;
int input[100][100];
struct node
{
int E,W;
node(int ee,int ww)
{
E=ee;W=ww;
}
node(){}
bool friend operator<(node n1,node n2)
{
return n1.W>n2.W;
}
};
void prim()
{
int dist[100];//未加入的点离当前已生成的最小生成树的最近的距离，次数足可以不要，加上以后可以优化队列大小
bool vis[100];//i点是否已经加入当前最小生成树
int ans=0;
priority_queue<node> Q;
memset(vis,false,sizeof(vis));
memset(dist,0x7f,sizeof(dist));
Q.push(node(0,0));//加入第一个点
while(!Q.empty())
{
node now=Q.top();
Q.pop();
if(!vis[now.E])//此点未加入
{
ans+=now.W;//更新当前已生成的最小生成树
vis[now.E]=true;
for(int i=0;i<N;i++)//检查从此点连出的边是否会更新未加入的点离已生成的树的距离
{
if(vis[i]) //从此点伸出的边相连的点已经在当前生成树中了
continue;
if(dist[i]<input[now.E][i])//从此点伸出的边到达未加入的点的距离大于以前加入的
continue;
dist[i]=input[now.E][i];
Q.push(node(i,input[now.E][i]));
}
}
}
cout<<ans<<endl;
}
int main()
{
freopen("agrinet.in","r",stdin);
freopen("agrinet.out","w",stdout);
scanf("%d",&N);
for(int i=0;i<N;i++)
{
for(int j=0;j<N;j++)
{
scanf("%d",&input[i][j]);
}
}
prim();
return 0;
}