hdu1233 还是畅通工程 prim

#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;

const int maxn = 105;
const int INF = 0xffffff;
struct Node
{
    int vex,weight;
    Node(int _vex = 0,int _weight = 0) : vex(_vex),weight(_weight){}
};
vector<Node> G[maxn];
bool intree[maxn];//intree[v] = true表示v在生成树集合中
int mindist[maxn],n;
//mindist[x] = y表示:
//x属于集合V,F中所有结点到x的最小距离为y,这里一定要好好理解!!!

void Init()
{
    for(int i = 0 ; i < maxn ; ++i){
        G[i].clear();
        intree[i] = false;
        mindist[i] = INF;//初始化为最大值,因为开始的时候F为空,到V的任意节点都不连通,用INF表示
    }
}
int Prim(int s)//s是第一个加入生成树的结点
{
    int ans = 0;
    int vex,addNode,tempMin;//加入集合F中的节点

    intree[s] = true;//加入生成树
    for(unsigned int i = 0 ; i < G[s].size() ; ++i){//更新mindistG[s][i].vex即所有与s连通的节点vex
        vex = G[s][i].vex;
        mindist[vex] = G[s][i].weight;                 //与s相连的点,这些点到s点的距离 (目前最小)
    }
    for(int nodeNum = 1 ; nodeNum <= n - 1 ; ++nodeNum){//对于集合V中剩余的n-1个节点
        tempMin = INF;
        /*******************************/
        for(int i = 1 ; i <= n ; ++i){                     //从集合V中找出到F最小距离的边,
            if(intree[i] == false && mindist[i] < tempMin){
                tempMin = mindist[i];
                addNode = i;
            }
        }
        /******************************/
        intree[addNode] = true;         //addNode 加入集合F
        ans += tempMin;                 //此时tempMin是最小权值
        /******************************/
        for(unsigned int i = 0 ; i < G[addNode].size() ; ++i){//addNode加入集合F后更新F到V的最小距离,这里最关键!
            vex = G[addNode][i].vex;
            if(intree[vex] == false && G[addNode][i].weight < mindist[vex])
                mindist[vex] = G[addNode][i].weight;
        }
        /*****************************/
    }
    return ans;
}
int main()
{
    int v1,v2,weight;

    while(scanf("%d",&n) != EOF && n){
        Init();
        for(int i = 0 ; i < n * (n - 1) / 2  ; ++i){
            scanf("%d%d%d",&v1,&v2,&weight);
            G[v1].push_back(Node(v2,weight));
            G[v2].push_back(Node(v1,weight));
        }
        int ans = Prim(1);//1结点作为起点
        printf("%d\n",ans);
    }

    return 0;
}

你可能感兴趣的:(C++,ACM,图论,HDU,Prim)