Acwing.342 道路与航线（堆dij + 拓扑排序）

最新推荐文章于 2023-02-10 20:25:21 发布

吃一口AC摇摇乐

最新推荐文章于 2023-02-10 20:25:21 发布

阅读量129

点赞数

分类专栏：图涂凸秃突轮论伦仑抡

本文链接：https://blog.csdn.net/lucifer1214/article/details/119386200

版权

图涂凸秃突轮论伦仑抡专栏收录该内容

20 篇文章 0 订阅

订阅专栏

思路：
首先题目给定混合图，双向边边权皆为正值，单向边有负值但无环。

看到有负值，那么我们就默认了 $d i j k s t r a$ 算法无法进行操作，但是经过 $s p f a$ 算法的尝试后，发现会被卡，所以我们需要进行进一步的思考。

$1 .$ 首先我们输入所有的双向边利用dfs将其形成连通块，我们将这样一个连通块称作一个团，并对其打上编号。再对每一个连通块内有那些点进行储存。

$2 .$ 输入单向边，并且对每一个连通块的入度进行计数。

$3 .$ 进行拓扑排序操作，首先将入度为 $0$ 的连通块入队，每次取出队头，对这个团中的所有点跑一遍 $d i j k s t r a$ 。再这个过程中的松弛操作需要进行修改，如果拓展到的点与该点不在一个团中我们，那么就说明我们遍历到了单向边，将对应团的出度 $- -$ ，并且入队，如果在一个团中就更新距离并且入堆。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<map>
#include<cstdio>
#include<cmath>
#include<iomanip>
#include<sstream>
#include<queue>

#define fr for
#define pi 3.1415926535
#define me(a,b,c) memset(a,b,sizeof c)
#define cut cout
#define eps 0.00000001
#define x first
#define y second

using namespace std;

typedef long long ll;
const int mod = 1e9 + 7;
typedef pair<int, int> pii;
const int N = 1e6 + 10;

int n, p, q, s;
int h[N], e[N], ne[N], w[N], idx;

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

int dis[N];
bool st[N];
queue<int>hh;
int id[N];
vector<int>block[N];
int bcnt;
int din[N];

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

void dijkstra(int bid) {
	priority_queue<pii, vector<pii>, greater<pii>>heap;

	for (auto u : block[bid])heap.push({ dis[u],u });

	while (heap.size())
	{
		auto t =  heap.top();
		heap.pop();
		int ver = t.y, dist = t.x;
		if (st[ver])continue;
		st[ver] = true;

		for (int i = h[ver]; ~i; i = ne[i]) {
			int j = e[i];
			if (id[j] != id[ver] && --din[id[j]] == 0)hh.push(id[j]);
			if (dis[j] > dis[ver] + w[i]) {
				dis[j] = dis[ver] + w[i];
				if (id[j] == id[ver])heap.push({ dis[j],j });
			}
		}
	}

}

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

	cin >> n >> p >> q >> s;
	memset(h, -1, sizeof h);
	for (int i = 1; i <= p; i++) {
		int a, b, c;
		cin >> a >> b >> c;
		add(a, b, c);
		add(b, a, c);
	}

	for (int i = 1; i <= n; i++) {
		if (!id[i]) {
			bcnt++;
			dfs(i, bcnt);
		}
	}
	
	for (int i = 1; i <= q; i++) {
		int a, b, c;
		cin >> a >> b >> c;
		din[id[b]]++;
		add(a, b, c);
	}

	topsort();

	for (int i = 1; i <= n; i++) {
		if (dis[i] > 0x3f3f3f3f / 2)puts("NO PATH");
		else cout << dis[i] << endl;
	}
	
}