洛谷P1195 口袋的天空 并查集

洛谷P1195 口袋的天空

并查集
将边从小到大排序，然后依次加边 看看是否能变成 K 个连通块，
以及其最小话费

 

#include <cstdio>
#include <algorithm> 
#define For(i,j,k) for(int i=j;i<=k;i++)  
using namespace std ; 

const int N = 1011,M = 10011,inf = 1e9 ;
struct edge{
    int x,y,val ; 
}e[M] ; 

int n,m,k,sum,num ; 
int fa[N] ; 

inline int read() 
{
    int x = 0 , f = 1 ; 
    char ch = getchar() ; 
    while(ch<'0'||ch>'9') { if(ch=='-') f = -1 ; ch = getchar() ; } 
    while(ch>='0'&&ch<='9') { x = x * 10+ch-48 ; ch = getchar() ; } 
    return x * f ;  
}

inline bool cmp(edge a,edge b) 
{
    return a.val < b.val ; 
}

inline int find(int x) 
{
    if(fa[x]==x) return x ; 
    return fa[ x ] = find(fa[x]) ; 
}

inline void work() 
{
    int x,y ; 
    num = 0 ; sum = 0 ; 
    sort(e+1,e+m+1,cmp) ; 
    For(i,1,m) {
        x = find(e[ i ].x) ; 
        y = find(e[ i ].y) ; 
        if(x!=y) {
            num++ ; sum+=e[ i ].val ; 
            if(num==n-k) {
                printf("%d\n",sum) ; 
                return ; 
            }
            fa[x] = y ; 
        }
    }
    printf("No Answer\n") ; 
}

int main() 
{
    while(~scanf("%d%d%d",&n,&m,&k)) {
        For(i,1,n) fa[ i ] = i ; 
        For(i,1,m) {
            e[ i ].x = read() ; e[ i ].y = read() ; e[ i ].val = read() ; 
        } 
        work() ; 
    }
    return 0 ; 
}