【codevs 1078】 最小生成树

题解：

        prim简单应用

        方法1：

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int inf=1<<30;

int n,ans=0,tot=1;
int dis[110],vis[110]={0};
int map[110][110];
int head[110];
struct edge{
	int v,w,next;
}e[100*100+10];

void readdata()
{
	memset(head,-1,sizeof(head));
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			scanf("%d",&map[i][j]);
}

void addedge(int u,int v,int w)
{
	e[tot].v=v;
	e[tot].w=w;
	e[tot].next=head[u];
	head[u]=tot++;
}

void prim()
{
	int min,k;
	for(int i=1;i<=n;i++)
		dis[i]=inf;
	dis[1]=0;
	int m=1;
	for(int i=1;i<=n;i++)
	{
		min=inf;
		for(int j=1;j<=n;j++)
		{
			if(dis[j]<min && !vis[j])
			{
				k=j;
				min=dis[j];
			}
		}
		ans+=min;
		vis[k]=1;
		addedge(m,k,min);
		m=k;
		for(int j=1;j<=n;j++)
		{
			if(map[k][j]<dis[j])
				dis[j]=map[k][j];
		}
	}
	printf("%d\n",ans);
}

int main()
{
	readdata();
	prim();
	return 0;
}

        方法2：图的矩阵存储+prim

#include <stdio.h>
#include <string.h>
const int inf=1<<30;
const int N=110;

//创建map二维数组储存图表，low数组记录每2个点间最小权值，vis数组标记某点是否已访问
int map[N][N],low[N],vis[N];
int n;

int prim()
{
	int i,j,pos,min,ans=0;
	memset(vis,0,sizeof(vis));
	//从某点开始，分别标记和记录该点
	vis[1]=1;pos=1;
	//第一次给low数组赋值
	for(i=1;i<=n;i++)
		if(i!=pos) low[i]=map[pos][i];
	//再运行n-1次
	for(i=1;i<n;i++)
	{
		//找出最小权值并记录位置
		min=inf;
		for(j=1;j<=n;j++)
			if(!vis[j] && min>low[j]) 
			{
				min=low[j];pos=j;
			}
		//最小权值累加
		ans+=min;
		//标记该点
		vis[pos]=1;
		//更新权值
		for(j=1;j<=n;j++)
			if(!vis[j] && low[j]>map[pos][j])
				low[j]=map[pos][j];
	}
	return ans;
}

int main()
{
	int i,j,ans;
	scanf("%d",&n);
	memset(map,inf,sizeof(map));
	for(i=1;i<=n;i++)
		for(j=1;j<=n;j++)
			scanf("%d",&map[i][j]);
	ans=prim();
	printf("%d\n",ans);
	return 0;
}