Dijkstra 模板(优先队列+vector存储图) O(mlog(n))

写成struct了

#include<iostream>
#include<queue>
#include<vector>
#include<algorithm>
const int maxn = 1007;
const int INF = 0x3f3f3f3f;
using namespace std;

struct Edge {
    int from, to, dist;
    Edge(int f, int t, int d):from(f), to(t), dist(d) {}
};
struct HeapNode {
    int u, d;
    bool operator < (const HeapNode& other) const {
        return d > other.d;
    }
};
struct Dijkstra {
    int n, m;
    vector<Edge> edges;//edges[i]存储了编号为i的边的具体信息
    vector<int> G[maxn];//G[u]是一个vector<int>,存储了节点u周围的边的编号
    bool done[maxn];
    int dis[maxn];//s到各个点的距离,即最终答案
    
    void init(int n) {
        this->n = n;
        for (int i=0;i<=n;i++) {
            done[i] = false;
            dis[i] = INF;
            G[i].clear();
        }
        edges.clear();
    }

    void AddEdge(int from, int to, int d) {
        edges.push_back(Edge(from, to, d));
        m = edges.size();
        G[from].push_back(m - 1);//把与from直接相连的边在edges里的标号放到G[from]里,由于此时是从末尾入的vector,故标号就是edges.size()-1
    }

    void dijkstra(int s) {
        dis[s] = 0;
        priority_queue<HeapNode> Q;
        Q.push(HeapNode{s, 0});

        while (!Q.empty()) {
            HeapNode x = Q.top();Q.pop();
            int& u = x.u;
            if (!done[u]) {
                for (int i=0;i<G[u].size();i++) {
                    Edge e = edges[G[u][i]];
                    if (dis[e.to] > dis[u] + e.dist) {
                        dis[e.to] = dis[u] + e.dist;
                        Q.push(HeapNode {e.to, dis[e.to]});
                    }
                }
                done[u] = true;
            }
        }
    }

};
int main()
{
    struct Dijkstra dj;
	int x, y, d, n, m;
	cin>>n>>m;
	dj.init(n);
	for (int i=1;i<=m;i++) {
		cin>>x>>y>>d;
		dj.AddEdge(x, y, d);
		dj.AddEdge(y, x, d);
	}
	dj.dijkstra(1);

	for (int i=1;i<=dj.n;i++) {
		cout<<dj.dis[i]<<" ";
	}cout<<endl;
    
    return 0;
}

然而本辣鸡还是不会输出最短的路径。。。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值