HDU 1879 继续畅通工程

题意

给定n个点和n*(n-1)/2条边的无向图，求图的最小生成树，其中有些边已经确定(可能是最小生成树的一部分，也可能不是)。

思路

由于已经确定了部分边，在读入数据过程中，如果这条边已经被确定，那我们就用并查集将两点合并；如果这条边还未确定，那么我们将它放进边集数组等待Kruskal算法进行从小到大的遍历即可。

const int N=110;
struct Node
{
    int a,b,c;
    bool operator<(const Node &W) const
    {
        return c < W.c;
    }
}e[N*N];
int p[N];
int n,m;

int find(int x)
{
    if(x != p[x]) p[x]=find(p[x]);
    return p[x];
}

int kruskal()
{
    sort(e,e+m);

    int res=0;
    for(int i=0;i<m;i++)
    {
        int a=e[i].a,b=e[i].b,c=e[i].c;
        int pa=find(a),pb=find(b);
        if(pa != pb)
        {
            p[pa]=pb;
            res+=c;
        }
    }
    return res;
}

int main()
{
    ios;
    while(cin>>n && n)
    {
        for(int i=1;i<=n;i++) p[i]=i;

        m=0;
        for(int i=0;i<n*(n-1)/2;i++)
        {
            int a,b,c,t;
            cin>>a>>b>>c>>t;
            if(t == 1)
            {
                int pa=find(a),pb=find(b);
                if(pa != pb) p[pa]=pb;
            }
            else e[m++]={a,b,c};
        }

        int t=kruskal();
        cout<<t<<endl;
    }

    //system("pause");
    return 0;
}