洛谷 [P3008] 道路与航线

最短路

因为有负权边,所以不能 dijkstra ,本题数据还卡 SPFA
但是我们发现,有负权的都是有向边,而且如果把无向边连成的联通块看成一个点的话,有向边就连成了一个 DAG,所以我们可以对所有的联通块用dij求最短路
在 DAG上用拓扑序求最短路

注意:
堆优化的 Dijkstra 在定义的结构体重载运算符的时候注意相反
因为存在负权边,所以两点不可达,不等价于两点间的距离 == inf ,应为 两点间的距离大于一个很大的数

#include <iostream>
#include <cmath>
#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <queue>
#include <vector>
#include <cstring>
using namespace std;
const int MAXN = 100005;
int init() {
    int rv = 0, fh = 1;
    char c = getchar();
    while(c < '0' || c > '9') {
        if(c == '-') fh = -1;
        c = getchar();
    }
    while(c >= '0' && c <= '9'){
        rv = (rv<<1) + (rv<<3) + c - '0';
        c = getchar();
    }
    return fh * rv;
}
int head[MAXN], n, m1, m2, s, nume, id[MAXN], tot, dis[MAXN], cnt[MAXN];
bool f[MAXN];
vector <int> ve[25005];
struct edge{
    int to, nxt, dis;
}e[MAXN<<1];
void adde(int from, int to, int dis) {
    e[++nume].to = to;
    e[nume].nxt = head[from];
    e[nume].dis = dis;
    head[from] = nume;
}
void dfs(int u) {
    if(id[u]) return;
    id[u] = tot;
    ve[tot].push_back(u);
    for(int i = head[u]; i; i = e[i].nxt) {
        int v = e[i].to;
        dfs(v);
    }
}
struct node{
    int num, val;
    bool operator < (const node & b) const{
        return val > b.val;
    }
};
priority_queue <node> q;
queue <int> qq;
void dij(int x){ 
    for(int i = 0; i < (int)ve[x].size(); i++) q.push((node){ve[x][i], dis[ve[x][i]]});
    while(!q.empty()) {
        node u = q.top(); q.pop();
        if(f[u.num]) continue;
        f[u.num] = 1;
        for(int i = head[u.num]; i; i = e[i].nxt) {
            node v;
            v.num = e[i].to;
            if(id[v.num] == id[u.num] && dis[v.num] > dis[u.num] + e[i].dis) {
                dis[v.num] = dis[u.num] + e[i].dis;
                v.val = dis[v.num];
                q.push(v);
            }
            if(id[v.num] != id[u.num]) {
                dis[v.num] = min(dis[v.num], dis[u.num] + e[i].dis);
                cnt[id[v.num]]--;
                if(!cnt[id[v.num]]) {
                    qq.push(id[v.num]);
                }
            }
        }
    }
}
int main() {
    n = init(); m1 = init(); m2 = init(); s = init();
    for(int i = 1; i <= m1; i++) {
        int u = init(), v = init(), di = init();
        adde(u, v, di); adde(v, u, di); 
    }
    for(int i = 1; i <= n; i++) {if(!id[i]) tot++;dfs(i);}
    for(int i = 1; i <= m2; i++) {
        int u = init(), v = init(), di = init();
        adde(u, v, di);cnt[id[v]]++;
    }
    memset(dis, 0x3f, sizeof(dis));
    dis[s] = 0;
    for(int i = 1; i <= tot; i++) if(!cnt[i]) qq.push(i);
    while(!qq.empty()) {
        int v = qq.front(); qq.pop();
        dij(v);
    }
    for(int i = 1; i <= n; i++) {
        if(dis[i] > 100000000) printf("NO PATH\n");
        else printf("%d\n", dis[i]);
    }
    return 0;
}

转载于:https://www.cnblogs.com/Mr-WolframsMgcBox/p/8660305.html