牛客-道路建设——最小生成树模板题

题目

题目链接

题目详解

• 就一非常简单的最小生成树问题而已，但是就是耗了我很长时间也不知错哪了(现在也不清楚)。。

我用的Kruskal算法实现最小生成树，而我是用的最小堆实现的，然后最小堆实现就硬是给我卡最后一个用例，我也不知道为什么？然后改普通数组排序就直接过了。。。

以下为最小堆实现的main函数，然后就硬是卡最后一个用例，把最小堆换数组存值后就过了。。

int main() {
    int c,n,m;
    int u,v,w;
    UnionFind uf(101);
    while (scanf("%d%d%d",&c,&n,&m)!=EOF){
        priority_queue<node>pq;
        while (n--){
            scanf("%d %d %d",&u,&v,&w);
            pq.push({u,v,w});
        }
        int cnt = m-1;
        int val = 0;
        while (!pq.empty()&&cnt>0){
            auto & t = pq.top();pq.pop();
            if(!uf.isconnected(t.u,t.v)){
                uf.merge(t.u,t.v);
                cnt--; val+=t.w;
            }
        }
        const char* str;
        if(cnt==0&&val<=c) str = "Yes";
        else str = "No";
        printf("%s\n",str);
        uf.clear();
    }
    return 0;
}

经验教训：竞赛OJ的存图题目可能有时候比较玄乎，比如我之前写的一道图论题目，是最短路的那种，然后他两个结点直接竟然存在多条边？我™。。。然后我就不得不用二维矩阵存图，而这次我猜又是这种类似的输入骚操作，导致我只能用数组存边。。

所以总结就是这种竞赛题，写最小生成树，最好使用数组存边！

解题代码

#include 
using namespace std;
using ll = long long;
//TODO 并查集实现
class UnionFind{
private:
    int* root;
    int* rank;
    int length;
public:
    UnionFind(int size):length(size){
        root = new int[size];
        rank = new int[size];
        for(int i=0;i<length;i++){
            root[i] = i;
            rank[i] = 1;
        }
    }
    int find(int x){
        if(root[x]==x)
            return x;
        return root[x]=find(root[x]);
    }
    void merge(int x,int y){
        int rootX = find(x);
        int rootY = find(y);
        //高度小的树被高度大的合并，如果高度一致合并后高度增加
        if(rootX!=rootY){
            if(rank[rootX]>rank[rootY]){
                root[rootY] = rootX;
            } else if(rank[rootX]<rank[rootY]){
                root[rootX] = rootY;
            }else{
                root[rootY] = rootX;
                rank[rootX]++;
            }
        }
    }
    bool isconnected(int x,int y){
        return find(x)==find(y);
    }
    void clear(){
        for(int i=0;i<length;i++){
            root[i] = i;
            rank[i] = 1;
        }
    }
};
//TODO 定义用于存边的结构
struct  node{
    int u,v,w;
    bool operator<(const node& a)const {
        return this->w<a.w;
    }
}edges[10005];

int main() {
    int c,n,m;
    int u,v,w;
    UnionFind uf(105);
    while (scanf("%d%d%d",&c,&n,&m)!=EOF){
        for(int i=0;i<n;i++){
            scanf("%d%d%d",&u,&v,&w);
            edges[i] = {u,v,w};
        }
        int cnt = m-1;
        int val = 0;
        sort(edges,edges+n);
        for (int i=0;i<n&&cnt>0;i++){
            node t = edges[i];
            if(!uf.isconnected(t.u,t.v)){
                uf.merge(t.u,t.v);
                cnt--; val+=t.w;
            }
        }
        const char* str;
        if(cnt==0&&val<=c) str = "Yes";
        else str = "No";
        printf("%s\n",str);
        uf.clear();
    }
    return 0;
}