NYOJ38布线问题_最小生成树

好久没做最小生成题了，只记得算法。。。快比赛了突然在这又出了问题

prim算法要用邻接矩阵适用于顶点少的题，做题习惯直接把这种用矩阵的先给排除了，直接用vector，结构体，导致一道题会用时很久

做题可不能像我这样，看到题要知道用什么算法，还要去想下需要用哪种数据结构（当算法比较多的时候，我们就要考虑题中的数据，根据题中给的数据大小选择数据结构，继而在选择这个算法的哪种）‘

因为顶点不多，就用了prim算法，用了krus（）但是runtime了。。。不想再改了~

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define inf 0x3f3f3f

using namespace std;
int n,m;
int map[505][505];
int dis[505];
int vis[505];
int prim()
{
memset(dis,0x3f,sizeof(dis));
memset(vis,0,sizeof(vis));
int i,count=0,u,v,sum=0;

for(i=1;i<=n;i++)
{
dis[i]=map[1][i];
}
vis[1]=1;
count++;
while(count<n)
{
int min=inf;
for(i=1;i<=n;i++)
{
if(!vis[i]&&dis[i]<min)
{
min=dis[i];
u=i;
}
}

vis[u]=1;
sum+=dis[u];
count++;

for(v=1;v<=n;v++)
{
if(!vis[v] && dis[v]>map[u][v])
{

dis[v]=map[u][v];
}
}
}

return sum;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
memset(map,0x3f,sizeof(map));

int i,u,v,w;
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++)
map[i][i]=0;
for(i=0;i<m;i++)
{
scanf("%d%d%d",&u,&v,&w);
map[u][v]=map[v][u]=w;
}

int min=1<<28;
for(i=0;i<n;i++)
{
scanf("%d",&u);
if(min>u)
{
min=u;
}
}

min+=prim();
printf("%d\n",min);

}
return 0;
}