PAT 最小路径题 Dijkstra+DFS模板

可用于求解最短路径问题,以及一些存在第二第三标尺,需要遍历路径的问题。

#include <cstdio>
#include <vector>
#include <cstring>
#include <algorithm>
#include <queue>
#include <functional>
using namespace std;

const int maxn = 1111, inf = 999999999;
int n;

int G[maxn][maxn], d[maxn];
bool vis[maxn] = { false };

typedef pair<int, int> P;
priority_queue<P, vector<P>, greater<P> > pq;

vector<int> pre[maxn], path, temp;

void Dijkstra(int s) {
    fill(d, d + maxn, inf);
    memset(vis, 0, sizeof(vis));

    P p;
    int u;
    cost[s] = 0;
    vis[s] = true;
    while (!pq.empty()) pq.empty();
    pq.push(P(0, s));
    while (!pq.empty()) {
        p = pq.top(); pq.pop();
        u = p.second;
        if (p.first > cost[u]) continue;

        vis[u] = true;
        for (int i = 0; i < n; i++) {
            if (vis[i] == false && G[u][i] != inf) {
                if (cost[i] > cost[u] + G[u][i]) {
                    cost[i] = cost[u] + G[u][i];
                    pq.push(P(cost[i], i));

                    pre[i].clear();
                    pre[i].push_back(u);

                }
                else if (cost[i] == cost[u] + G[u][i]) {
                    pre[i].push_back(u);
                }
            }
        }
    }
}

void DFS(int v) {
    if (v == from) {
        temp.push_back(v);
        //此处定义第二,第三标尺变量
        for (int i = temp.size() - 1; i > 0; i--) {
        // 根据题设判断是计算点权还是计算边权,稍稍有些微变化
        }
        //第二第三标尺判断
        temp.pop_back();
        return;
    }

    temp.push_back(v);
    for (int i = 0; i < pre[v].size(); i++) {
        DFS(pre[v][i]);
    }
    temp.pop_back();
}

int main() {
    fill(G[0], G[0] + maxn*maxn, inf);
    // TODO:
}

    原文作者:Dijkstra算法
    原文地址: https://blog.csdn.net/fried123123/article/details/82292081
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞