CodeForces 1307D BFS最短路思维

_int_me

已于 2024-01-22 00:35:45 修改

阅读量183

点赞数

分类专栏：图论应用贪心、思维相关文章标签：图论算法 c++

于 2021-02-07 19:03:58 首次发布

本文链接：https://blog.csdn.net/intmeX/article/details/113745842

版权

贪心、思维相关同时被 2 个专栏收录

6 篇文章 0 订阅

订阅专栏

图论应用

5 篇文章 0 订阅

订阅专栏

本文介绍了一种解决无向图中最短路径问题的策略，当边权均为1且存在特殊节点时，如何通过增加特定边来最大化从1到n的最短路径。通过使用BFS算法计算每个节点到1和n的最短距离，并利用排序和一维枚举优化，实现了低于n^2的时间复杂度解决方案。最后，给出了AC代码实现。

摘要由CSDN通过智能技术生成

原题链接

题意

给出一个简单无向图，边权全部为1，同时给我们k个特殊点，要求我们从这k个特殊点中选出两个来连一条边权为1的边。同时，我们的决策要保证1~n的最短路程最大，求最终这个最短路长度。

思路

首先因为边权都为1，所以我们可以BFS来求出最短路。而且可以顺带求出所有点的d1(1到i的路程)和d2(i到n的路程)
然后考虑如果连接i，j点，那么最终结果就是 $min (d 1 [i] + d 2 [j] + 1, d 1 [n])$ , 不过要注意的是，这里必须有 $\le d1[j] - d2[j]$ ,否则结果不符合最短路的定义。
因为n很大，所以我们需要一个低于n^2时间复杂度的算法，那么首先我们要将二维枚举优化为一维。这里可以看到

$\le d1[j] + d2[i]$

可以化为

$\le d1[j] - d2[j]$

那么我们可以按照这个变形后的式子来对点进行排序。然后可知，后面的点的d2总是可以和前面的点的d1配对，所以我们从前往后扫描，同时用同一个变量来维护d1的最大值即可得到答案。

AC代码

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>

using namespace std;

const int N = 200000;

int n, k, m;
int fir[N + 5], nex[N * 2 + 5], vv[N * 2 + 5];
bool fl[N + 5];
struct ab
{
	int v[2];
	int l;
	bool operator < (const struct ab& c) const
	{
		return v[0] - v[1] < c.v[0] - c.v[1];
	}
} dis[N + 5];
int ff[N + 5];

void bfs(int s, int l)
{
	queue<int> qq;
	qq.push(s);
	dis[s].v[l] = 0;
	while (!qq.empty())
	{
		int cc = qq.front();
		qq.pop();
		for (int i = fir[cc]; i; i = nex[i])
		{
			if (dis[vv[i]].v[l] == -1)
			{
				dis[vv[i]].v[l] = dis[cc].v[l] + 1;
				qq.push(vv[i]);
			}
		}
	}
}

int main()
{
	scanf("%d%d%d", &n, &m, &k);
	memset(dis, -1, sizeof(dis));
	for (int i = 1; i <= n; ++i)
	{
		dis[i].l = i;
		ff[i] = i;
	}
	for (int i = 1; i <= k; ++i)
	{
		int x;
		scanf("%d", &x);
		fl[x] = true;
	}
	for (int i = 1; i <= m; ++i)
	{
		scanf("%d%d", &vv[i + m], &vv[i]);
		nex[i] = fir[vv[i + m]];
		fir[vv[i + m]] = i;
		nex[i + m] = fir[vv[i]];
		fir[vv[i]] = i + m;
	}
	bfs(1, 0);
	bfs(n, 1);
	int cc = dis[n].v[0];
	sort(dis + 1, dis + 1 + n);
	int ans = 0;
	int l1 = 1;
	while (!fl[dis[l1].l])
	{
		++l1;
	}
	int mx = dis[l1].v[0];
	for (int i = l1 + 1; i <= n; ++i)
	{
		if (fl[dis[i].l])
		{
			ans = max(mx + dis[i].v[1], ans);
			mx = max(dis[i].v[0], mx);
		}
	}
	printf("%d\n", min(ans + 1, cc));
	return 0;
}