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

Posted 2022-02-05 C+++++++++++++++++++

文章目录

• • 题目
  • 题目详解
  • 解题代码

题目

题目链接

题目详解

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

我用的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 <bits/stdc++.h>
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;