【挖沟——Kruskal】

挖沟

---

题目描述

胡队长带领HA实验的战士们玩真人CS，真人CS的地图由一些据点组成，现在胡队长已经占领了n个据点，为了方便，将他们编号为1-n，为了隐蔽，胡队长命令战士们在每个据点出挖一个坑，让战士们躲在坑里。由于需要在任意两个点之间传递信息，两个坑之间必须挖出至少一条通路，而挖沟是一件很麻烦的差事，所以胡队长希望挖出数量尽可能少的沟，使得任意两个据点之间有至少一条通路，顺便，尽可能的∑d[i][j]使最小（其中d[i][j]为据点i到j的距离）。

---

输入描述:

第一行有2个正整数n，m，m表示可供挖的沟数。
接下来m行，每行3个数a,b,v，每行描述一条可供挖的沟，该沟可以使a与b连通，长度为v。

输出描述:

输出一行，一个正整数，表示要使得任意两个据点之间有一条通路，至少需要挖长的沟。（数据保证有解）

示例

示例1：
输入
2 2
1 2 1
1 2 3

输出
1

示例2
输入
3 3
1 2 3
2 3 4
1 3 5

输出
7

备注:
对于100%的测试数据：
1 ≤ n ≤ 100000
1 ≤ m ≤ 500000
1 ≤ v ≤ 10000

---

AC代码

代码如下：

#include<bits/stdc++.h>
using namespace std;
const int N=500005;
struct node{
    int a,b,v;
};node point[N];
int n,m,f[N];
bool cmp(node x,node y){
    return x.v<y.v;
}
void init(){
    for(int i=1;i<=n;++i)
        f[i]=i;
}
int find(int x){
    if(x!=f[x])
        f[x]=find(f[x]);
    return f[x];
}
bool mix(int x,int y){
    int f_x=find(x);
    int f_y=find(y);
    if(f_x==f_y)
        return false;
    f[f_x]=f_y;
    return true;
}
int Kruskal(){
    int ans=0,cnt=0;
    init();
    for(int i=0;i<m-1;++i)
        if(mix(point[i].a,point[i].b)){
            ans+=point[i].v;
            ++cnt;
        }
    return ans;
}
int main(){
    cin>>n>>m;
    for(int i=0;i<m;++i)
        cin>>point[i].a>>point[i].b>>point[i].v;
    sort(point,point+m,cmp);
    cout<<Kruskal()<<endl;
}

---

总结

最小生成树模板题，kruskal和prim都可以实现