【最短路 SPFA】JZOJ_3086 洛谷_3831 [SHOI2012]回家的路

最新推荐文章于 2020-04-02 23:47:09 发布

nymph181

最新推荐文章于 2020-04-02 23:47:09 发布

阅读量188

点赞数

分类专栏：图论文章标签：图论最短路 SPFA

本文链接：https://blog.csdn.net/SSL_hzb/article/details/86618145

版权

图论专栏收录该内容

67 篇文章 0 订阅

订阅专栏

题意

给出一个 $N * N$ 的方格图，这里有 $N$ 横 $N$ 纵条线路，其中有 $M$ 个中转站。

坐车每做一站需要 $2$ 分钟，中途转车需要 $1$ 分钟，求出从起点到终点的最短时间。如果不能从站内转车到终点，输出 $- 1$ 。

思路

考试时看到这题以为是动态规划，然后发现好像不准确，就让它做了 $2$ 遍，发现样例都能过，然后拿到了 $30$ 分。

正解是最短路。

把每个点拆成横点和纵点，它们之间连一条边，代价为 $1$ 表示转车。

对于相同横坐标或纵坐标的点，也把它们连起来，代价就为坐车的时间。

跑一边 $S P F A$ 就好了，也可以考虑 $D i j k s t r a$ 。

代码

#include<queue>
#include<vector>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define px(i) (i) * 2 - 1
#define py(i) (i) * 2

std::vector<int> inX[20001], inY[20001];
struct node{
	int to, next, w;
}e[480009];
int n, m, ans, tot;
int x[100003], y[100003], d[200005], head[200005], v[200005];

void add(int u, int v, int w) {
	e[++tot] = (node){v, head[u], w};
	head[u] = tot;
	e[++tot] = (node){u, head[v], w};
	head[v] = tot;
}

int spfa(int start, int end) {
	memset(d, 127 / 3, sizeof(d));
	std::queue<int> q;
	q.push(px(start));q.push(py(start));
	v[px(start)] = v[py(start)] = 1;
	d[px(start)] = d[py(start)] = 0;
	while (q.size()) {
		int u = q.front();
		q.pop();
		v[u] = 0;
		for (int i = head[u]; i; i = e[i].next) {
			int to = e[i].to;
			if (d[u] + e[i].w < d[to]) {
				d[to] = d[u] + e[i].w;
				if (!v[to]) {
					q.push(to);
					v[to] = 1;
				}
			}
		}
	}
	return std::min(d[px(end)], d[py(end)]) < 707406378 ? std::min(d[px(end)], d[py(end)]) : -1;
}

int cmpX(int a, int b) {
	return y[a] < y[b];
}

int cmpY(int a, int b) {
	return x[a] < x[b];
}

int main() {
	scanf("%d %d", &n, &m);
	m += 2;
	for (int i = 1; i <= m; i++) {
		scanf("%d %d", &x[i], &y[i]);
		add(px(i), py(i), 1);
		add(py(i), px(i), 1);
		inX[x[i]].push_back(i);
		inY[y[i]].push_back(i);
	}
	for (int i = 1; i <= n; i++) {
		std::sort(inX[i].begin(), inX[i].end(), cmpX);
		std::sort(inY[i].begin(), inY[i].end(), cmpY);//排序为了让相邻的两个点相连
		for (int j = 0; j + 1 < inX[i].size(); j++) {
			add(px(inX[i][j]), px(inX[i][j + 1]), (y[inX[i][j + 1]] - y[inX[i][j]]) * 2);
			add(px(inX[i][j + 1]), px(inX[i][j]), (y[inX[i][j + 1]] - y[inX[i][j]]) * 2);
		}
		for (int j = 0; j + 1 < inY[i].size(); j++) {
			add(py(inY[i][j]), py(inY[i][j + 1]), (x[inY[i][j + 1]] - x[inY[i][j]]) * 2);
			add(py(inY[i][j + 1]), py(inY[i][j]), (x[inY[i][j + 1]] - x[inY[i][j]]) * 2);
		}
	}
	printf("%d", spfa(m - 1, m));
}