【牛客赛前训练6】 路径难题

一、题目内容

牛牛最近在为自己的出行烦忧。

牛牛所在的城市可以简化为 $n$ 个点，编号从 $1$ 开始。由 $m$ 条无向边表达两两之间的路径。第 $i$ 条边连通点 $u_i$ 和 $v_i$，距离为$d_i$。

牛牛是个懒人，他宁愿叫出租车也不愿意走路。牛牛所在城市的出租车不计算起步价，直接按照距离计费，每 $r$ 单位距离 $1$ 元，向上取整，即如果单次行进了 $p$ 单位距离，将收费⌈$\frac{p}{r}$⌉（向上取整）元。如果多次乘坐分开收费。

牛牛所在的城市还有 $k$ 路不同的公交车，每一路公交车都是双向的，有 $t_i$
个站台，可以正着坐，也可以反着坐，但每路公交车仅会在规定的站台停车。牛牛从任意一个位置上车，可以从该路公交车的任意站台下车，仅需付一份该路的公交车车费 $c_i$ 元。但坐到终点站之后必须下车。

牛牛有 $q$ 个规划，其中第i个规划为从 $a_i$ 位置移动到 $b_i$ 位置，牛牛想知道这样的移动最小需要的花费是多少。花费是指出租车的花费和公交车的花费之和。

二、题目分析

$Subtask$ $1$ ：

$k=0$ ，此时直接跑一遍 $Dijkstra$ 求解 $a_i$ 到 $b_i$ 的最短路即可，期望得分 $20-40$ (不能跑 $Floyd$， 因为此题可能有重边和自环）（然后我考试时写了个完美的 $Floyd$ 爆炸了）

$Subtask$ $2$ ：

首先可以考虑建一个分层图，如图所示：

显然建图的代价为 $O(n^2)$，这是无法承受的，（然而我考试时也没想到更好的方法，就只写了暴力，然后暴力还写错了……）注意到公交停的站台之间相互可达且代价为 $0$ ， 我们可以建立一个虚点，将每条公交车的站台的点与一个虚点分别连一条边（边权为 $r\times c_i$（原因后面解释）），再将这个虚点与每个为站台的点之间连接一条边权为 $0$ 的边， 如图所示：

然后再对此图跑 $Dijkstra$， 时间复杂度为 $O(q(n+m)logn)$

关于 $SPFA$ ——他死了（得分75）

(此题中考虑到出租车每下车一次要结算费用（向上取整），如果将其存储起来比较繁琐，而公交车的费用是已知的，那么我们将费用乘以 $r$ 后即可转化为距离，然后每次访问到公交车的虚点时我们就把前面的距离向上变为 $6$ 的倍数，这样保证了最终的答案也是向上取整的，统一向虚点建时可以更好地对上公交车时的前置距离处理）

三、参考代码

链式前向星存图：（777ms）

#include <queue>
#include <cstdio>
#include <cctype>
#include <algorithm>
#define int long long
const int N = 8e5 + 10, INF = 2e15;
struct Node {
	int now, val;
	bool operator<(const Node &a) const{return val > a.val;}
};
struct Edge {int to, val, nxt;}e[N << 1];
int head[N], num, n, m, k, q, tot, vis[N], r, dis[N];
std::priority_queue <Node> pq;
template <typename T>inline void rd(T &x) {
	x = 0; T f = 1; char ch = getchar();
	for (; !isdigit(ch); ch = getchar()) if (ch == '-') f = -1;
	for (; isdigit(ch); ch = getchar()) x = (x * 10) + (ch ^ 48);
	x = x * f;
}
template <typename T>inline void write(T x) {
	if (x < 0) putchar('-'), x = -x;
	if (x > 9) write(x / 10);
	putchar(x % 10 + 48);
}
inline void AddEdge(const int &u, const int &v, const int &w) {
	e[++num] = (Edge){v, w, head[u]}, head[u] = num;
}
inline int Make(const int &x, const int &y) {return x / y + (x % y != 0);}

inline void dijkstra(const int &s) {
	for (int i = 1; i <= tot; ++i) dis[i] = INF, vis[i] = 0;
	pq.push((Node){s, 0}); dis[s] = 0;
	while (!pq.empty()) {
		int u = pq.top().now; pq.pop();
		if (vis[u]) continue; vis[u] = 1;
		for (int i = head[u]; i; i = e[i].nxt) {
			int v = e[i].to, w = e[i].val;
			if (dis[v] > dis[u] + w) {
				if (v > n) dis[v] = Make(dis[u], r) * r + w;
				else dis[v] = dis[u] + w;
				pq.push((Node){v, dis[v]});
			}
		}
	}
}
signed main(void) {
	rd(n), tot = n, rd(m), rd(k), rd(r), rd(q);
	for (int i = 1, u, v, w; i <= m; ++i) {
		rd(u), rd(v), rd(w);
		AddEdge(u, v, w), AddEdge(v, u, w);
	}
	for (int i = 1, t, c; i <= k; ++i) {
		rd(t); rd(c); ++tot;
		for (int j = 1, u; j <= t; ++j)
			rd(u), AddEdge(u, tot, c * r), AddEdge(tot, u, 0);
	}
	for (int u, v; q; --q) {
		rd(u), rd(v);
		dijkstra(u);
		write(Make(dis[v], r)); putchar('\n');
	}
	return 0;
}

$vector$ 存图：（706ms）

#include <queue>
#include <cstdio>
#include <cctype>
#include <algorithm>
#define int long long
const int N = 8e5 + 10, INF = 2e15;
struct Node {
	int now, val;
	bool operator<(const Node &a) const{return val > a.val;}
};
struct Edge {int to, val;};
int head[N], num, n, m, k, q, tot, vis[N], r, dis[N];
std::vector <Edge> e[N];
std::priority_queue <Node> pq;
template <typename T>inline void rd(T &x) {
	x = 0; T f = 1; char ch = getchar();
	for (; !isdigit(ch); ch = getchar()) if (ch == '-') f = -1;
	for (; isdigit(ch); ch = getchar()) x = (x * 10) + (ch ^ 48);
	x = x * f;
}
template <typename T>inline void write(T x) {
	if (x < 0) putchar('-'), x = -x;
	if (x > 9) write(x / 10);
	putchar(x % 10 + 48);
}
inline void AddEdge(const int &u, const int &v, const int &w) {
	e[u].push_back((Edge){v, w});
}
inline int Make(const int &x, const int &y) {return x / y + (x % y != 0);}

inline void dijkstra(const int &s) {
	for (int i = 1; i <= tot; ++i) dis[i] = INF, vis[i] = 0;
	pq.push((Node){s, 0}); dis[s] = 0;
	while (!pq.empty()) {
		int u = pq.top().now; pq.pop();
		if (vis[u]) continue; vis[u] = 1;
		int sz = e[u].size();
		for (int i = 0; i < sz; ++i) {
			int v = e[u][i].to, w = e[u][i].val;
			if (dis[v] > dis[u] + w) {
				if (v > n) dis[v] = Make(dis[u], r) * r + w;
				else dis[v] = dis[u] + w;
				pq.push((Node){v, dis[v]});
			}
		}
	}
}
signed main(void) {
	rd(n), tot = n, rd(m), rd(k), rd(r), rd(q);
	for (int i = 1, u, v, w; i <= m; ++i) {
		rd(u), rd(v), rd(w);
		AddEdge(u, v, w), AddEdge(v, u, w);
	}
	for (int i = 1, t, c; i <= k; ++i) {
		rd(t); rd(c); ++tot;
		for (int j = 1, u; j <= t; ++j)
			rd(u), AddEdge(u, tot, c * r), AddEdge(tot, u, 0);
	}
	for (int u, v; q; --q) {
		rd(u), rd(v);
		dijkstra(u);
		write(Make(dis[v], r)); putchar('\n');
	}
	return 0;

}