POJ 2421 Constructing Roads Kruskal求解

題目鏈接：http://acm.hust.edu.cn/vjudge/contest/view.action?cid=66965#problem/D

題意：給定一些村莊及邊，然後給出一些，已經存在的邊，求最小生成樹。

思路：題意簡單，根據Kruskal思路，先把已經存在的邊加進去，然後就是原樣的Kruskal了。

代碼：

#include<iostream>
#include<cstdio>
#include<algorithm>

using namespace std;
struct edge{
    int u,v,w;
    bool operator <(const edge &a)const{
        return w<a.w;
    }
}E[10010];

int N,Q,tot,pre[110];

int find(int x){
    int t=x;
    while(t!=pre[t]) t=pre[t];
    while(x!=t) pre[x]=t,x=pre[x];
    return t;
}

void Kruskal(){
    int sum=0;
    for(int i=0;i<tot;i++){
        int u=E[i].u,v=E[i].v,w=E[i].w;
        int fu=find(u),fv=find(v);
        if(fu==fv) continue;
        pre[fv]=fu;sum+=w;
    }
    cout<<sum<<endl;
}

int main(){
    //freopen("D:\\in.txt","r",stdin);
    while(cin>>N){
        int t;
        tot=0;
        for(int i=0;i<N;i++) for(int j=0;j<N;j++){
            scanf("%d",&t);
            if(j<i) E[tot++]=(edge){i,j,t};
        }
        cin>>Q;
        for(int i=0;i<=N;i++) pre[i]=i;
        int a,b;
        for(int i=0;i<Q;i++){
            scanf("%d %d",&a,&b);
            pre[find(a-1)]=find(b-1);
        }
        sort(E,E+tot);
        Kruskal();
    }
    return 0;

}