欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

数算 兔子与星空 图最小支撑树prim算法

程序员文章站 2022-07-13 07:58:16
...

兔子与星空

题目内容:很久很久以前,森林里住着一群兔子。兔子们无聊的时候就喜欢研究星座。如图所示,天空中已经有了n颗星星,其中有些星星有边相连。兔子们希望删除掉一些边,然后使得保留下的边仍能是n颗星星连通。他们希望计算,保留的边的权值之和最小是多少?

输入格式:
第一行只包含一个表示星星个数的数n,n不大于26,并且这n个星星是由大写字母表里的前n个字母表示。接下来的n-1行是由字母表的前n-1个字母开头。最后一个星星表示的字母不用输入。对于每一行,以每个星星表示的字母开头,然后后面跟着一个数字,表示有多少条边可以从这个星星到后面字母表中的星星。如果k是大于0,表示该行后面会表示k条边的k个数据。每条边的数据是由表示连接到另一端星星的字母和该边的权值组成。权值是正整数的并且小于100。该行的所有数据字段分隔单一空白。该星星网络将始终连接所有的星星。该星星网络将永远不会超过75条边。没有任何一个星星会有超过15条的边连接到其他星星(之前或之后的字母)。

输出格式:
输出是一个整数,表示最小的权值和

输入样例:
9
A 2 B 12 I 25
B 3 C 10 H 40 I 8
C 2 D 18 G 55
D 1 E 44
E 2 F 60 G 38
F 0
G 1 H 35
H 1 I 35

输出样例:
216

#include <iostream>
#include <cstring>
#include <string>
#include <vector>
using namespace std;

int a[30][30] = { 0 };
int visited[30] = { 0 };
vector<int> MST;

int prim(int n)
{
 int v = 1;
 int result = 0;
 MST.push_back(v);
 visited[v] = 1;
 while (1)
 {
  int size = MST.size();
  if (size == n)
   break;
  int minval = 9999;
  int minnumber;
  for (int i = 0; i < size; i++)
 {
   int cur = MST[i];
   for (int j = 1; j <= n; j++)
   {
    if (visited[j] == 0 && a[cur][j] != 0 && a[cur][j] < minval)
    {
     minval = a[cur][j];
     minnumber = j;
    }
   }
  }
  //cout << minnumber << " " << minval << endl;调试
  result += minval;
  MST.push_back(minnumber);
  visited[minnumber] = 1;
 }
 return result;
}

int main()
{
 int n;
 cin >> n;
 for (int i = 1; i < n; i++)
 {
  int m;
  int num;
  char c; 
  cin >> c;
  cin >> m;
  while (m--)
  {
   cin >> c >> num;
   a[i][c - 'A' + 1] = num;
   a[c - 'A' + 1][i] = num;
  }
 }
 cout << prim(n) << endl;
 return 0;
}
相关标签: 数算