堆优化Dijkstra 模版

#include <cstdio>
#include <algorithm>
using namespace std;
const int inf=1000000000;
int n,m,top,idx,s;
int to[2000001],next[2000001],len[2000001],head[100001],
    h[131073],pos[100001],d[100001];
bool v[100001];
void add(int x,int y,int z)
{
    to[++idx]=y;
    len[idx]=z;
    next[idx]=head[x];
    head[x]=idx;

    to[++idx]=x;
    len[idx]=z;
    next[idx]=head[y];
    head[y]=idx;
    return ;
}
void pushup(int x)
{
    while(x>1)
    {
        if(d[h[x]]<d[h[x>>1]])
            swap(h[x],h[x>>1]),
            swap(pos[h[x]],pos[h[x>>1]]),
            x>>=1;
        else
            break;
    }
    return ;
}
void insert(int x)
{
    h[++top]=x;
    pos[x]=top;
    pushup(top);
    return ;
}
void pop()
{
    pos[h[1]]=0;
    h[1]=h[top--];
    if(top)
        pos[h[1]]=1;
    int x=2;
    while(x<=top)
    {
        if(d[h[x]]>d[h[x+1]])
            ++x;
        if(d[h[x]]<d[h[x>>1]])
            swap(h[x],h[x>>1]),
            swap(pos[h[x]],pos[h[x>>1]]),
            x<<=1;
        else
            break;
    }
    return ;
}
void dijkstra()
{
    for(int i=1;i<=n;++i)
        d[i]=inf;
    d[s]=0;
    insert(s);
    while(top)
    {
        int x=h[1];pop();
        if(v[x])
            continue;
        v[x]=true;
        for(int i=head[x];i;i=next[i])
            if(!v[to[i]]&&d[to[i]]>d[x]+len[i])
            {
                d[to[i]]=d[x]+len[i];
                insert(to[i]);
            }
    }
    return ;
}
int main()
{
    s=1;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;++i)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
    }
    dijkstra();
    printf("%d",d[n]);
    return 0;
}
下面是一个使用优化Dijkstra算法的模板: ```cpp #include <iostream> #include <vector> #include <queue> #include <limits> using namespace std; typedef pair<int, int> pii; // 存储节点和距离的pair const int INF = numeric_limits<int>::max(); // 无穷大 vector<vector<pii>> graph; // 图的邻接表表示 vector<int> dijkstra(int start, int n) { vector<int> dist(n, INF); // 存储起始点到每个点的最短距离 dist[start] = 0; priority_queue<pii, vector<pii>, greater<pii>> pq; // 小顶,存储节点和距离的pair pq.push(make_pair(0, start)); // 将起始点入队 while (!pq.empty()) { int u = pq.top().second; // 当前节点 int d = pq.top().first; // 当前距离 pq.pop(); // 如果当前距离大于已经计算出的最短距离,则忽略 if (d > dist[u]) continue; // 遍历当前节点的邻接节点 for (auto edge : graph[u]) { int v = edge.first; // 邻接节点 int w = edge.second; // 边的权重 // 如果通过u到v的距离更短,则更新最短距离,并将v加入队列 if (dist[u] + w < dist[v]) { dist[v] = dist[u] + w; pq.push(make_pair(dist[v], v)); } } } return dist; } int main() { int n, m; cin >> n >> m; // 输入节点数和边数 graph.resize(n); for (int i = 0; i < m; i++) { int u, v, w; cin >> u >> v >> w; // 输入边的起点、终点和权重 graph[u].push_back(make_pair(v, w)); graph[v].push_back(make_pair(u, w)); // 无向图需要双向连接 } int start; cin >> start; // 输入起始点 vector<int> dist = dijkstra(start, n); // 调用Dijkstra算法 // 输出最短距离 for (int i = 0; i < n; i++) cout << "Node " << i << ": " << dist[i] << endl; return 0; } ``` 这个模板使用了优化的方法来实现Dijkstra算法,通过优先队列(小顶)来选择当前距离最短的节点进行扩展。算法的时间复杂度为O((V+E)logV),其中V为节点数,E为边数。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值