数据结构考试最小生成树

题目描述
根据输入创建无向网。分别用Prim算法和Kruskal算法构建最小生成树。（假设：输入数据的最小生成树唯一。）

输入
顶点数n

n个顶点

边数m

m条边信息,格式为：顶点1 顶点2 权值

Prim算法的起点v

输出
输出最小生成树的权值之和

对两种算法，按树的生长顺序，输出边信息(Kruskal中边顶点按数组序号升序输出)

样例输入
6
v1 v2 v3 v4 v5 v6
10
v1 v2 6
v1 v3 1
v1 v4 5
v2 v3 5
v2 v5 3
v3 v4 5
v3 v5 6
v3 v6 4
v4 v6 2
v5 v6 6
v1
样例输出
15
prim:
v1 v3 1
v3 v6 4
v6 v4 2
v3 v2 5
v2 v5 3
kruskal:
v1 v3 1
v4 v6 2
v2 v5 3
v3 v6 4
v2 v3 5

#include <iostream>
#include <map>
#include <sstream>
#include <queue>

using namespace std;

const int maxn = 105;
const int INF = 0x3f3f3f3f;

int Graph[maxn][maxn];

int pre[maxn];

int dis[maxn];

bool vis[maxn];

int from[maxn];

int n;
    
map<string,int> name;

int find(int x)
{
    return x == pre[x] ? x : pre[x] = find(pre[x]);
}

void combine(int x, int y)
{
    int px = find(x), py = find(y);
    pre[px] = py;
}

string getname(int i)
{
    for(map<string, int>::iterator it = name.begin(); it != name.end(); it++)
        if(it -> second == i)
            return it -> first;
}

void init()
{
    name.clear();
    for(int i = 0; i < maxn; i++)
    {
        from[i] = i;
        pre[i] = i;
        vis[i] = false;
        dis[i] = INF;
        for(int j = 0; j < maxn; j++)
            Graph[i][j] = INF;
            
    }
}

void prim(int s)
{
    
    vis[s] = true;
    
    stringstream ss;
    int sum = 0;
    
    for(int i = 0; i < n; i++)
        if(!vis[i] &&dis[i] > Graph[s][i])
            dis[i] = Graph[s][i], from[i] = s;

    
    for(int N = 1; N < n; N++)
    {
        int minn = INF, mini = -1;
        for(int i = 0; i < n; i++)
            if(minn > dis[i] && !vis[i])
                minn = dis[i], mini = i;

        vis[mini] = true;
        ss << getname(from[mini]) << ' ' << getname(mini) << ' ' << Graph[from[mini]][mini] << endl;
        sum += Graph[mini][from[mini]];
        
        for(int i = 0; i < n; i++)
            if(!vis[i] && dis[i] > Graph[mini][i])
                dis[i] = Graph[mini][i], from[i] = mini;
        
    }
    
    cout << sum << endl;
    cout << "prim:" << endl;
    
    cout << ss.str();
    
}

void kruskal()
{
    cout << "kruskal:" << endl;
    struct Node
    {
        int u, v, w;
        bool operator < (const Node &x) const
        {
            return w > x.w;
        }
    };
    
    priority_queue<Node> pq;
    
    for(int i = 0; i < n; i++)
        for(int j = i; j < n; j++)
            if(Graph[i][j] != INF)
                pq.push(Node{i, j, Graph[i][j]});
                
    while(!pq.empty())
    {
        Node tem = pq.top();
        pq.pop();
        
        if(find(tem.u) != find(tem.v))
        {
            cout << getname(tem.u) << ' ' << getname(tem.v) << ' ' << tem.w << endl;
            combine(tem.u, tem.v);
        }
    }
}

int main()
{

    init();
    
    cin >> n;
    
    for(int i = 0; i < n; i++)
    {
        string tem;
        cin >> tem;
        name[tem] = i;
    }
    
    int m;
    cin >> m;
    
    
    
    for(int i = 0; i < m; i++)
    {
        string u, v;
        int w;
        cin >> u >> v >> w;
        
        Graph[name[u]][name[v]] = Graph[name[v]][name[u]] = w;
    }
    
    string s;
    cin >> s;
    prim(name[s]);
    kruskal();

}