Poj 1258 Agri-Net

[题目链接]

思路：好多都是村村通啊，再水最后一个好了~

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int inf=0x3f3f3f3f;
const int Max_v=110;
const int Max_e=11000;

struct edge{
    int from,to,cost;
    bool operator<(const edge& e)const{
        return cost<e.cost;
    }
}e[Max_e];
int V,E;
int par[Max_v];
int cost[Max_v][Max_v];

int Find(int x){
    if(par[x]==x)return x;
    return par[x]=Find(par[x]);
}

int Kruskal(){
    int ans=0,sum=0;
    for(int i=0;i<V;i++)par[i]=i;
    sort(e,e+E);
    for(int i=0;i<E;i++){
        int x=Find(e[i].from);
        int y=Find(e[i].to);
        if(x!=y){
            ans++;sum+=e[i].cost;
            par[x]=y;
        }
        if(ans==V-1)break;
    }
    return sum;
}

int main()
{
    while(~scanf("%d",&V)){
        int c;
        for(int i=0;i<V;i++){
            for(int j=0;j<V;j++){
                scanf("%d",&c);
                cost[i][j]=c;
            }
        }
        E=0;
        for(int i=0;i<V;i++){
            for(int j=i+1;j<V;j++){
                e[E].from=i;e[E].to=j;
                e[E++].cost=cost[i][j];
            }
        }
        printf("%d\n",Kruskal());
    }
    return 0;
}