一个菜狗的悲伤
查看原帖
一个菜狗的悲伤
1157431
Houjiadong楼主2024/3/14 16:15
#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;
typedef pair<int, int> PII;
const int N = 1e5+10;
int n, m,s;
int h[N], w[2 * N + 10], e[N], ne[N], idx;
int dist[N];
bool st[N];
void add(int a, int b, int c) {
    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx++;
}
void dijkstra() {
    memset(dist, 0x3f, sizeof dist);
    dist[s] = 0;
    priority_queue<PII, vector<PII>, greater<PII>> heap;
    heap.push({ 0,s });
    while (heap.size()) {
        auto t = heap.top();
        heap.pop();
        int ver = t.second, distance = t.first;
        if (st[ver]) continue;
        st[ver] = true;
        for (int i = h[ver];i != -1;i = ne[i]) {
            int j = e[i];
            if (dist[j] > dist[ver] + w[i]) {
                dist[j] = dist[ver] + w[i];
                heap.push({ dist[j],j });
            }
        }
    }
}
int main() {
    cin >> n >> m >> s;
    memset(h, -1, sizeof(h));
    while (m--) {
        int a, b, c;
        cin >> a >> b >> c;
        add(a, b, c);
    }
    dijkstra();
    for (int i = 1;i <= n;i++) {
        cout << dist[i] <<" ";
    }
}

哎有两个测试点WA了 目前还是不会判断哪里数据越界,以及各种情况的dijkstra算法怎么用 只是单纯背模板罢了

2024/3/14 16:15
加载中...