hdu1102 最小生成树 特殊处理

<span style="font-family: Arial, Helvetica, sans-serif; background-color: rgb(255, 255, 255);">该题是最小生成树的变形，因为给出的是任何两个点之间的距离，且其中某些路已经修理，问额外修理的路的总长是多少</span>

如果按最小生成树的算法来算，比如 ，我用的是prim算法，刚开始使用的时候，我的想法是这样，用used[]数组来储存某点是否已经用过，如果a和b点已经修过了路，则used[a]和used[b]都赋值为1，然后把两点间的距离设置为0，再把两点push到优先队列中。

而这种想法是错的，因为即时所有的点都used了，但是可能会出现，所有点不是一个连通图，比如共4个点，可能会出现1-2，3-4这样的连法。

所以最后又想了一下， 如果已经修路，只把两条路间的距离设置为0即可，这样就能保证，在计算的时候，能把这两点的那条路给计算进去。

贴代码：

#include <cstdio>
#include <iostream>
#include <utility>
#include <queue>
#include <vector>
#include <cstring>
#include <string>

using namespace std;

int N;//N 个村庄
const int V_MAX = 105;
typedef pair<int, int> P;
int cost[V_MAX][V_MAX];
vector<int> G[V_MAX];
int used[V_MAX];

int prim()
{
    int sum = 0;
    int Q;
    priority_queue<P, vector<P>, greater<P> > pque;
    scanf("%d", &Q);
    for(int i=1; i<=Q; i++)
    {
        int fir, sec;
        scanf("%d%d", &fir, &sec);
        cost[fir][sec] = 0;
        cost[sec][fir] = 0;
    }
    pque.push(P(0,1));
    while(!pque.empty()){
        P temp = pque.top();
        pque.pop();

        int V = temp.second;
        int cos = temp.first;
        if(used[V]!=1){
            sum += cos;
            used[V] = 1;
        }
        for(int i=1; i<=N; i++)
        {
            if((used[i]!=1)&&(V!=i))
                pque.push(P(cost[V][i],i));
        }
    }
    return sum;
}
int main()
{
    while(scanf("%d", &N)!=EOF)
    {
        memset(used, 0, sizeof(used));
        memset(G, 0, sizeof(G));
        int temp;
        for(int i=1; i<=N; i++)
        {
            for(int j=1; j<=N; j++)
            {
                scanf("%d",&cost[i][j]);
            }
        }
        printf("%d\n",prim());
    }
    return 0;
}