1212 无向图最小生成树
N个点M条边的无向连通图,每条边有一个权值,求该图的最小生成树。
Input
第1行:2个数N,M中间用空格分隔,N为点的数量,M为边的数量。(2 <= N <= 1000, 1 <= M <= 50000)
第2 - M + 1行:每行3个数S E W,分别表示M条边的2个顶点及权值。(1 <= S, E <= N,1 <= W <= 10000)
Output
输出最小生成树的所有边的权值之和。
Input示例
9 14
1 2 4
2 3 8
3 4 7
4 5 9
5 6 10
6 7 2
7 8 1
8 9 7
2 8 11
3 9 2
7 9 6
3 6 4
4 6 14
1 8 8
Output示例
37
注意:
1.数组开的大小,如果数组太大的话,就会内存超时。
2.最小生成树算法,细节注意的点在代码中
#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<math.h>
using namespace std;
#define INF 0x3f3f3f3f
const int maxn=1100;
int n,m;
int mp[1100][1100];
int dis[maxn];
int vis[maxn];
int prime()
{
int minn,temp,sum=0;
for(int i=1;i<n;i++)
{
minn=INF;
for(int j=1;j<=n;j++)
{
if(vis[j]==0 && dis[j]<minn)//需要注意 dis[j]<minn
{
minn=dis[j];
temp=j;
}
}
vis[temp]=1;
sum+=minn;
for(int j=1;j<=n;j++)
{
if(vis[j]==0 && mp[temp][j]<dis[j])
{
dis[j]=mp[temp][j];//需要注意 mp[temp][j]<dis[j]
}
}
}
return sum;
}
int main()
{
scanf("%d%d",&n,&m);
memset(vis,0,sizeof(vis));
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
if(i==j)
{
mp[i][j]=0;
}
else
{
mp[i][j]=INF;
}
}
}
for(int i=0;i<m;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
if(mp[u][v]>w)
{
mp[u][v]=w;
mp[v][u]=w;
}
}
for(int i=0;i<=n;i++)
{
dis[i]=mp[1][i];//需要注意 mp[1][i]
}
vis[1]=1;
cout<<prime()<<endl;
return 0;
}