ssl2346-联络员【图论，最小生成树】

题目

一个图，给几条固定的边，求最小生成树

输入

第一行n，m表示n个点，有m条边，每行四个非负整数，p,u,v,w 当p=1时，表示这个边为固定边（必须要）；当p=2时，表示一条边；u,v,w表示点u，v之间的边，w表示权值。
5 6
1 1 2 1
1 2 3 1
1 3 4 1
1 4 1 1
2 2 5 10
2 2 5 5

输出

最小生成树的值和固定边的值的和
9

---

解题思路

这道题用O(n log n)的算法，把固定边链接的点看成同一个点。

---

代码

#include<cstdio>
#include<algorithm>//算法库
using namespace std;
struct wolf//表示一条线
{
    int cost,first,last;
};
int n,h,k,s,w,x,y,q,lt[2001],num;
wolf a[10001];//线
bool cmp(wolf x,wolf y)
{
    return x.cost<y.cost;
}//排序比较器
int father(int x)
{
    if (lt[x]!=x) lt[x]=father(lt[x]);
    return lt[x];//找初始点（并查集）
}
void unionn(int x,int y)
{
    int fa=father(x);
    int fb=father(y);
    if (fa!=fb) lt[fa]=fb;
}//改变点
int main()
{
    scanf("%d%d",&n,&h);
    for (int i=1;i<=n;i++) lt[i]=i;//初始化联通分量
    for (int i=1;i<=h;i++)
    {
        scanf("%d%d%d%d",&q,&x,&y,&w);
        if (q==1)
        {
            unionn(x,y);//合并两个点
            s+=w;//记录固定边的值
        }
        else
        {
            k++;
            a[k].cost=w;
            a[k].first=x;
            a[k].last=y;//边
        }
    }
    sort(a+1,a+1+k,cmp);//排序
    for (int i=1;i<=k;i++)
    {
        if (father(a[i].first)!=father(a[i].last))
        //如果联通分量不一样
        {
            s+=a[i].cost;//记录权值
            num++;//记录一条边
            unionn(a[i].first,a[i].last);//合并
        }
        if (num==n-1) break;//如果边数够设则退出
    }
    printf("%d",s);
}