题目

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

输入

第一行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);
}