关闭

hdu1102 最小生成树 特殊处理

标签: 1102最小生成树
219人阅读 评论(0) 收藏 举报
分类:

<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;
}



0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:6970次
    • 积分:269
    • 等级:
    • 排名:千里之外
    • 原创:18篇
    • 转载:10篇
    • 译文:0篇
    • 评论:0条
    文章分类
    文章存档