首页 > 其他 > 详细

Dijkstra + heap

时间:2014-07-28 00:07:00      阅读:443      评论:0      收藏:0      [点我收藏+]

用优先队列实现,和朴素Dijkstra差不多,核心思想没有改变。

模板:

struct node    
{
    int pos, dist;
    friend bool operator < (node a, node b)  // 定义优先级
    {
        return a.dist > b.dist;   // 小的先出
    }
};
int Dijkstra(int n,int start,int end)  // n个点,定义起点和终点
{
    bool visited[maxn];
    int pos;
    node tmp;
    tmp.pos = start;
    tmp.dist = 0;
    priority_queue<node> q;
    for(int i = 1; i <= n; i++)
    {
        visited[i] = 0;
        dist[i] = INF;
    }
    dist[start]= 0;
    q.push(tmp);   // 将起点插入队列
    while(!q.empty())
    {
        tmp = q.top();
        q.pop();
        pos = tmp.pos;
        if(tmp.dist == INF)  // 没有边直接返回
            return INF;
        if(visited[pos])
            continue;
        visited[pos] = 1;  
        for(int i = 1; i <= n; i++)
            if(!visited[i] && tmp.dist + map[pos][i] < dist[i])
            {
                dist[i] = tmp.dist + map[pos][i];
                node t; t.pos = i; t.dist = dist[i];
                q.push(t);       // 将更新后的边插入队列
            }
    }
    return dist[end];
}


Dijkstra + heap,布布扣,bubuko.com

Dijkstra + heap

原文:http://blog.csdn.net/userluoxuan/article/details/38173089

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!