【最小生成树-Prim】HDU 1863 畅通工程

Prim:(O(V^2))

• 与Kruskal不同，Prim是跑点
• 用dis[ i ]来维护结点 i 到树的最小路径
• 如果我们已经用该点更新过其他点的dis[ ]就标记为1，标记过的点不会再被拿来更新别的点，所以我们的树上不会有环
• 可能所给条件不能构成最小生成树，所以要判断一下

步骤：

1. 初始化dis[ ]为INF，起点的dis为0
2. 然后我们跑一个for循环从未标记的点中找到最小的dis[ ]，标记位置为min_pos
3. 更新从min_pos出发的点的dis[ ]
4. 重复2、3操作，直到跑完V个点为止

HDU 1863 畅通工程

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cmath>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <stack>
#include <list>
#include <map>
#define P(x) x>0?x:0
#define INF 0x3f3f3f3f

using namespace std;
typedef long long ll;
typedef vector<int>:: iterator VITer;
const int maxV=105;
const int maxE=105;
int E,V;

int cnt,head[maxV];
struct EDGE{
    int pre,to,w;
    EDGE(int a=0,int b=0,int c=0):pre(a),to(b),w(c){}
    friend bool operator < (EDGE e1,EDGE e2) { return e1.w<e2.w; }
}edge[maxE*2];
void add_edge(int u,int v,int val)
{
    edge[cnt]=EDGE(head[u],v,val);
    head[u]=cnt++;
}
void init()
{
    memset(head,-1, sizeof(head));
    cnt=0;
}

int dis[maxV],vis[maxV];//每个点到树上的最小路径
int Prim()//跑点
{
    int ans=0;
    memset(vis,0, sizeof(vis));
    for(int i=1;i<=V;i++)
        dis[i]=(i==1 ? 0 : INF);
    for(int i=1;i<=V;i++)
    {
        int min_pos=0,minn=INF;
        for(int j=1;j<=V;j++)
        {
            if(!vis[j] && dis[j]<minn)
            {
                minn=dis[j];
                min_pos=j;
            }
        }
        vis[min_pos]=1;
        if(minn==INF)
            return -1;
        ans+=minn;
        for(int j=head[min_pos]; ~j ; j=edge[j].pre)
        {
            if(dis[edge[j].to]>edge[j].w)
            {
                dis[edge[j].to]=edge[j].w;
            }
        }
    }
    return ans;
}
int main()
{
    while(~scanf("%d%d",&E,&V) && E)
    {
        init();
        for(int i=0;i<E;i++)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            add_edge(a,b,c);
            add_edge(b,a,c);
        }
        if(E<V-1||Prim()==-1)
        {
            printf("?\n");
            continue;
        }
        printf("%d\n",Prim());
    }
    return 0;
}