poj-3723【图的最小生成树+并查集】

这个题目做的时候知道是最小生成树，用的kruskal算法，结果在并查集的地方出问题了。

刚开始的same函数弄得不对，，，还好，这个题目算是又复习了一下并查集

 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <set>
#include <vector>
#include <climits>
#include <queue>
using namespace std;
typedef long long LL;
const int maxn = 50060;
int n,m,r;

struct node
{
    int x;
    int y;
    int z;
};
node edge[maxn];
int dx[20005];
int Find(int num1)
{
    if (dx[num1]==num1)
    {
        return num1;
    }
    else
    {
       return  dx[num1]=Find(dx[num1]);
    }
}
bool same(int num1,int num2)//same当中时find，不是直接的dx[num1]==dx[num2],这是不对的
{

    num1=Find(num1);
    num2=Find(num2);
    if (num1==num2)return true;
    else
        return false;
}
bool Unoin(int num1,int num2)//合并当中时树的合并，所以合并的是两个树的根
{
    num1=Find(num1);
    num2=Find(num2);
    dx[num1]=num2;
}
bool cmp(node it,node is)
{
    return it.z<is.z;
}
void Find()
{
    sort(edge,edge+r,cmp);
    int all=(n+m)*10000;
    for (int i=0;i<r;i++)
    {
        if (!same(edge[i].x,edge[i].y))
        {
            all+=edge[i].z;
            Unoin(edge[i].x,edge[i].y);
        }

    }
    printf("%d\n",all);
}
int main()
{
    int T;
    cin>>T;
    int num1,num2,num3;
    while(T--)
    {
        scanf("%d %d %d",&n,&m,&r);
        for (int i=0;i<r;i++)
        {
            scanf("%d %d %d",&num1,&num2,&num3);
            edge[i].x=num1+1,edge[i].y=num2+n+1,edge[i].z=-num3;
        }
        for (int i=0;i<=n+m;i++)
        {
            dx[i]=i;
        }
        Find();
    }
    return 0;
}

 

转载于:https://www.cnblogs.com/GregZQ/p/8365299.html