最小生成树(kruskal算法)

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
int n,m;
const int maxn = 1e5;
int f[maxn];
struct node
{
    int x;int y;int d;
}tmp[maxn];
int cmp(node a,node b)
{
    return a.d < b.d;
}
int getf(int x)
{
    if(f[x] == x) return x;
    else  return (f[x] = getf(f[x]));
}
int mix(int x,int y)
{
    int t1 = getf(x);
    int t2 = getf(y);
    if(t1 != t2)
    {
        f[t1] = t2;
        return 1;
    }
    return 0;
}
int main()
{
    cin >> n >> m;
    for(int i = 0;i <= n;i++)
        f[i] = i;
    for(int i = 0;i < m;i++)
        cin >> tmp[i].x >> tmp[i].y >> tmp[i].d;
    sort(tmp,tmp+m,cmp);
    long long sum = 0;
    for(int i = 0;i < m;i++)///算法的核心部分
    {
        if(mix(tmp[i].x,tmp[i].y))
            sum += tmp[i].d;
    }
    cout << sum << endl;
    return 0;
}
/**
6 9
2 4 11
3 5 13
4 6 3
5 6 4
2 3 6
4 5 7
1 2 1
3 4 9
1 3 2
**/

算法的核心部分其实还可以优化,就是当边检验到n-1的时候,就已经成了一棵树了,就不需要继续进行判断的,可以省下一部分判断的时间

long long sum = 0;
 int cnt  = 0;
    for(int i = 0;i < m;i++)
    {
        if(mix(tmp[i].x,tmp[i].y))
            sum += tmp[i].d,cnt++;
            if(cnt == n-1)
                break;
    }



你可能感兴趣的:(树)