zoj 1372 Networking

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=372

解题思路：最小生成树（kruskal 算法 + 优先队列 + 并查集）

/**************************************************************************
user_id: SCNU20102200088
problem_id: zoj 1372
problem_name: Networking
**************************************************************************/

#include <algorithm>
#include <iostream>
#include <iterator>
#include <iomanip>
#include <sstream>
#include <fstream>
#include <cstring>
#include <cstdlib>
#include <climits>
#include <bitset>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <ctime>
#include <cmath>
#include <queue>
#include <stack>
#include <list>
#include <set>
#include <map>
using namespace std;

//线段树
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

//手工扩展栈
#pragma comment(linker,"/STACK:102400000,102400000")

const double EPS=1e-9;
const double PI=acos(-1.0);
const double E=2.7182818284590452353602874713526;  //自然对数底数
const double R=0.5772156649015328606065120900824;  //欧拉常数:(1+1/2+...+1/n)-ln(n)

const int x4[]={-1,0,1,0};
const int y4[]={0,1,0,-1};
const int x8[]={-1,-1,0,1,1,1,0,-1};
const int y8[]={0,1,1,1,0,-1,-1,-1};

typedef long long LL;

typedef int T;
T max(T a,T b){ return a>b? a:b; }
T min(T a,T b){ return a<b? a:b; }
T gcd(T a,T b){ return b==0? a:gcd(b,a%b); }
T lcm(T a,T b){ return a/gcd(a,b)*b; }

///////////////////////////////////////////////////////////////////////////
//Add Code:
int n,s[55];

struct Node{
    int u,v,w;
    bool operator <(const Node &a) const{
        return w>a.w;
    }
};

priority_queue<Node> pq;

void Union(int x,int y){
    s[y]=x;
}

int Find(int x){
    if(s[x]<0) return x;
    return s[x]=Find(s[x]);
}

int Kruskal(){
    int ret=0,num=0;
    memset(s,-1,sizeof(s));
    while(!pq.empty() && num<n-1){
        Node res=pq.top();
        pq.pop();
        if(Find(res.u)!=Find(res.v)){
            Union(Find(res.u),Find(res.v));
            ret+=res.w;
            num++;
        }
    }
    while(!pq.empty()) pq.pop();
    return ret;
}
///////////////////////////////////////////////////////////////////////////

int main(){
    std::ios::sync_with_stdio(false);
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    ///////////////////////////////////////////////////////////////////////
    //Add Code:
    int m;
    while(scanf("%d",&n)!=EOF){
        if(n==0) break;
        scanf("%d",&m);
        while(m--){
            Node node;
            scanf("%d%d%d",&node.u,&node.v,&node.w);
            pq.push(node);
        }
        int ans=Kruskal();
        printf("%d\n",ans);
    }
    ///////////////////////////////////////////////////////////////////////
    return 0;
}

/**************************************************************************
Testcase:
Input:
1 0

2 3
1 2 37
2 1 17
1 2 68

3 7
1 2 19
2 3 11
3 1 7
1 3 5
2 3 89
3 1 91
1 2 32

5 7
1 2 5
2 3 7
2 4 8
4 5 11
3 5 10
1 5 6
4 2 12

0
Output:
0
17
16
26
**************************************************************************/