hdu 1012 constructing the road

最小生成树
村庄联通，但是已经有修好的路了，那么就是说已经修好的路，是已经存在的，不用修的，换句话说，已经修好的路的长度不在最小生成树的总长度里，且最小生成树必定包含这些已经修好的路。

根据最小生成树的算法prim ，如果把这些修好的路村庄节点之间的距离设为0,那么这些修好的路，就一定会被选中，且由于距离设为0,最后总长度，也不包括这些修好路的村庄之间的距离。

思路： 首先把已经存在路的村庄节点距离设为0.然后按照prim算法，构造最小生成树。

prim算法
prim 算法的思路如下：
首先用集合V表示所有节点的集合。mst集合表示，被选到最小生成树的节点的集合。
1. 随机选一个节点，放到mst mst={v1}， 令 visited[v1]=true 表示v1已经在mst里
2. 遍历mst中的每一个节点的连接点，找到最小边的另一个节点v2 mst={v1,v2} visited[2]=true
3. 重复2 (找v1,v2连接点的最小边的节点) ，直到mst中包含了所有节点。退出。

// 除了传统的循环查找最小值外，可以利用优先队列这种数据结构，priprity_queue

代码：

void handle(){

    vector<int> mst;

    memset(visited,false,sizeof(visited));
    mst.clear();
    int minimum=0;
    int start=1;

    mst.push_back(start);
    visited[start]=true;
    int len=mst.size();

    int ans=INF;
    int index=0;
// 以下是重点（while loop）
    while(mst.size()!=n){
        ans=INF;
        index=0;
        for(int i=0;i<mst.size();i++){

            // find the minimum dist j
            for(int j=1;j<=n;j++){

                int v=g[mst[i]][j];
                if(!visited[j]&&v<ans){
                    ans=v;
                    index=j;
                }
            }
        }
        mst.push_back(index);
       // cout<<"the index is: "<<index<<endl;
        visited[index]=true;

    }
}

AC 代码：

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


#define MAX   105
#define INF   100000000
int g[MAX][MAX];

bool visited[MAX];
int n;
int q;

void handle(){

    vector<int> mst;

    memset(visited,false,sizeof(visited));
    mst.clear();
    int minimum=0;
    int start=1;

    mst.push_back(start);
    visited[start]=true;
    int len=mst.size();

    int ans=INF;
    int index=0;
    while(mst.size()!=n){

        ans=INF;
        index=0;
        for(int i=0;i<mst.size();i++){

            // find the minimum dist j
            for(int j=1;j<=n;j++){

                int v=g[mst[i]][j];
                if(!visited[j]&&v<ans){

                    ans=v;
                    index=j;
                }

            }
        }

        mst.push_back(index);
       // cout<<"the index is: "<<index<<endl;
        visited[index]=true;
        minimum+=ans;
    }


    cout<<minimum<<endl;

}


int main()
{
    while(cin>>n){



        memset(g,INF,sizeof(g));
        for(int i=1;i<=n;i++){

            for(int j=1;j<=n;j++){
                cin>>g[i][j];
            }
        }

        cin>>q;
        int a,b;

        // a,b is connect no need road
        for(int k=1;k<=q;k++){

            cin>>a>>b;

            g[a][b]=0;
            g[b][a]=0;
        }

        //
        handle();
    }

    return 0;
}