数据结构与算法实验题 10.9 最大生成树

数据结构与算法实验题 10.9 最大生成树

★实验任务

问题很简单，众所周知，最小生成树是指生成树中权重和最小的树。

小布是一个爱思考的小孩，他学习完最小生成树之后，在思考另一个问题最大生成树，现在给你 N 个点（1 到 N），及各点间的关系，要求帮小布算出 N 个点的最大生成树,即所有生成树中边权和最大的。(注意两点间可能有多条边)

★数据输入

输入第一行包括两个正整数 N（N<=1000），M（M<100000），表示点数。

接下来 M 行，每行包括三个整数A, B, C，表示点 A 与点 B 相连，权重是 C(0<C<1000)，两个数之间空格隔开。

★数据输出

如果存在生成树，则输出最大生成树，如果无法构建生成树，则输出-1。

输入示例

5  6
1 2 1
1 2 1
2 3 1
3 4 1
4 5 1
5 1 1000

输出示例

1003

 思路：

可以直接使用Prim或者Kruskal算法，只需要更改排序的顺序即可，本人使用Kruskal算法。程序会自动更新连通块的数量（可使用并查集实现）

实现代码

#include <cstdio> 
#include <algorithm> 
#include <vector>
using namespace std;
#define maxint 0x7fffffff
struct line{
    int x, y;
    int len;
};
bool cmp(const line a,const line b){
    if(a.len>b.len)
        return true;
    return false;
}
bool isSame(int u,int v,vector<vector<int>> &tree,int n){
    int root1 = maxint;
    int root2 = maxint;
    for (int i = 1; i <= n;i++){
        if(find(tree[i].begin(),tree[i].end(),u)!=tree[i].end()){
            root1 = i;
        }
        if(find(tree[i].begin(),tree[i].end(),v)!=tree[i].end()){
            root2 = i;
        }
    }
    if(root1!=root2){
        for (int i = 0; i < tree[root2].size();i++){
            tree[root1].push_back(tree[root2][i]);
        }
        tree[root2].clear();
        return true;

    }
        return false;
}
line branch[100001];
int main()
{
    int n, m;
    int count=0;//用于统计连通块个数，即树的数量
    int sum = 0;
    //int x, y, len;
    scanf("%d%d", &n, &m);
    count = n;//初始连通块数量（均为孤立点）
    for (int i = 0; i < m;i++){
        scanf("%d%d%d", &branch[i].x, &branch[i].y, &branch[i].len);
    }
    sort(branch, branch + m, cmp);
    vector<vector<int>> tree(1001);//vector二维数组的正确写法
    for (int i = 1; i <= n;i++){
        tree[i].push_back(i);
    }
        for (int i = 0; i < m; i++)
        {
            int u = branch[i].x;
            int v = branch[i].y;
            if(isSame(u,v,tree,n)){
                count--;
                sum += branch[i].len;
                if(count==1)
                    break;//经过测试这一句话可以带来很大的时间提升
            }
        }
    if(count!=1)
        sum = -1;
    printf("%d", sum);
    return 0;
}