1863 畅通工程-并查集最小生成树

题目链接

问题描述：

　　

简单的最小生成树的题，将路径按cost从小到大排序，利用克鲁斯塔尔求最小生成树算法就行。

代码：

　　

#include<iostream>
#include<queue>
#include<algorithm>

using namespace std;
struct  Road
{
    int beg;
    int end;
    int cost;
    Road(int beg, int end, int cost)
    {
        this->beg = beg;
        this->end = end;
        this->cost = cost;
    }
    bool operator <(const Road& b) const 
    //必须要带const，使用STL，或者自己写compare函数
    //优先级大的在队列的前面，所以重载的时候需要注意需要的形式
    {
        return (cost > b.cost);
    }
    
};

int father[101];
int getfather(int a) //路径压缩
{
    if(father[a] != a)
    {
            father[a] = getfather(father[a]);
        return father[a];
    }
    else
        return father[a];
}

priority_queue<Road> q;
//并查集

int num;
int vnum;

int main()
{
    cin>>num>>vnum;
    int scost = 0;
    int  ok = 0;
    Road a = {0,0,0};
    while(num)
    {
        scost = 0;
        for(int i=0; i<=vnum; i++)
        {
            father[i] = i;
        }
        for(int i=0; i<num; i++)
        {
            cin>>a.beg>>a.end>>a.cost;
            q.push(a);
        }
        while(!q.empty())
        {
            a = q.top();
            q.pop();
            if(getfather(a.beg) != getfather(a.end))
            {
                father[a.end] = father[a.beg];
                scost += a.cost;
                ok++;
            }
        }
        if(ok == vnum-1)
            cout<<scost<<endl;
        else
            cout<<"?"<<endl;
        ok = 0;
        cin>>num>>vnum;

    }
    return 0;
}