prim算法
#include <stdio.h>
#include <string.h>
#define N 401
int state[N][N];
int maps [N][N];
int solve(int n)
{
int weight[N];
int adjex[N];
int visit[N];
int i,j,k;
int minnum,ans;
ans=0;
for(i=0;i<n;i++)
weight[i]=maps[0][i],adjex[i]=visit[i]=0;
visit[0]=1;
for(i=1;i<n;i++)
{
minnum=0xffffff;
for(j=1;j<n;j++)
if(minnum>weight[j] && visit[j] == 0)
minnum=weight[j],k=j;
visit[k]=1;
ans+=minnum;
for(j=1;j<n;j++)
if(maps[k][j]<weight[j] && visit[j]==0 )
weight[j]=maps[k][j],adjex[j]=k;
}
return ans;
}
int main()
{
int i,j,k,n;
while( scanf("%d",&n)!=EOF)
{
for(i=0;i<n;i++)
for(j=0;j<n;j++)
scanf("%d",&maps[i][j]);
printf("%d\n",solve(n));
}
return 0;
}
模板题
题目大意:首先给你一个图,需要你求出最小生成树,输入N个节点,用大写字母表示了节点,然后节点与节点之间有权值。
比如有9个节点,然后接下来有n-1行表示了边的情况,拿第一行举例
A 2 B 12 I 25
表示A有两个邻点,B和I,AB权值是12,AI权值是25
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int maxn = 30;
const int inf = 0x3f3f3f3f;
int road[maxn][maxn], dis[maxn];
bool vis[maxn];
int n;
void prim()
{
int mi, v;
for(int i = 0; i < n; i++)
{
dis[i] = road[0][i];
vis[i] = false;
}
for(int i = 0; i < n; i++)//包括第一个点在内,一共要纳入n个点
{
mi = inf;
for(int j = 0; j < n; j++)//每次找出未纳入顶点集与已知顶点集构成的权值最小的一条边
{
if(!vis[j] && mi > dis[j])
{
v = j;
mi = dis[j];
}
}
vis[v] = true;//把该顶点纳入已知集合
for(int j = 0; j < n; j++)//更新与未纳入集合中的顶点的边的最小权值
{
if(!vis[j] && dis[j] > road[v][j])
dis[j] = road[v][j];
}
}
for(int i = 1; i < n; i++)
dis[0] += dis[i];
cout << dis[0] << endl;
}
int main()
{
int m, w;
char s[5], en[5];
while(cin >> n, n)
{
memset(road, 63, sizeof(road));
for(int i = 0; i < n; i++)
road[i][i] = 0;
for(int i = 1; i < n; i++)
{
scanf("%s%d", s, &m);
for(int j = 0; j < m; j++)
{
scanf("%s%d", en, &w);
road[s[0]-'A'][en[0]-'A'] = road[en[0]-'A'][s[0]-'A'] = w;
}
}
prim();
}
return 0;
}