Acwing 342. 道路与航线

题目大意

有m1条道路 都为双向路 且边权大于0 有m2条航线 都为单向路 边权可能小于0 给出起点S 问从S到各点的最小距离为多少 若不存在则输出 “NO PATH”

题目思路

本来是想用spfa直接从起点搜最短路的 结果恶心的是这个题把spfa卡掉了
所以只能用dijkstra来搜了 但是dijkstra不能带有搜负权边的图
题中不难发现只有单向路即航线的边权可能为负权 而道路的都为正权边
所以我们可以把道路输入后找连通块然后建立拓扑图
按拓扑序 每次dijkstra 都在各自的连通块上进行
如果搜到了下一个连通块 不放入堆中 只把自己连通块中的点放入堆中

代码

#include <bits/stdc++.h>

using namespace std;

typedef pair<int, int> PII;

const int N = 25010, M = 150010, INF = 0x3f3f3f3f;

int n, m1, m2, S;
int h[N], e[M], w[M], ne[M], idx;
int id[N], bcnt, dist[N], din[N];
bool st[N];

vector<int> blocks[N];
queue<int> q;

void add(int a, int b, int c)
{
    e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx ++ ;
}

void dfs(int u, int bid)
{
    id[u] = bid, blocks[bid].push_back(u);
    
    for (int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if (!id[j]) dfs(j, bid);
    }
}

void dijkstra(int u)
{
    priority_queue<PII, vector<PII>, greater<PII>> heap;
    
    for (auto t : blocks[u]) heap.push({dist[t], t});
    
    while (heap.size())
    {
        auto t = heap.top();
        heap.pop();
        int ver = t.second;
        
        if (st[ver]) continue;
        st[ver] = true;
        
        for (int i = h[ver]; ~i; i = ne[i])
        {
            int j = e[i];
            if (dist[j] > dist[ver] + w[i])
            {
                dist[j] = dist[ver] + w[i];
                if (id[j] == id[ver]) heap.push({dist[j], j});
            }
            if (id[j] != id[ver] && -- din[id[j]] == 0) q.push(id[j]);
        }
    }
}

void topsort()
{
    memset(dist, 0x3f, sizeof dist);
    dist[S] = 0;
    
    for (int i = 1; i <= bcnt; i ++ )
        if (!din[i]) q.push(i);
    
    while (q.size())
    {
        int t = q.front();
        q.pop();
        dijkstra(t);
    }
}

int main()
{
    scanf("%d%d%d%d", &n, &m1, &m2, &S);
    
    memset(h, -1, sizeof h);
    
    while (m1 -- )
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        add(a, b, c), add(b, a, c);
    }
    
    for (int i = 1; i <= n; i ++ )
        if (!id[i]) dfs(i, ++ bcnt);
            
    while (m2 -- )
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        add(a, b, c), din[id[b]] ++ ;
    }
    
    topsort();
    
    for (int i = 1; i <= n; i ++ )
        if (dist[i] > INF / 2) puts("NO PATH");
        else printf("%d\n", dist[i]);
    
    return 0;
}