【ACWing】3797. 最大化最短路

题目地址：

https://www.acwing.com/problem/content/3800/

给定一个$n$个点$m$条边的无向连通图。图中所有点的编号为$1\sim n$。图中不含重边和自环。指定图中的$k$个点为特殊点。现在，你必须选择两个特殊点，并在这两个点之间增加一条边。所选两点之间允许原本就存在边。我们希望，在增边操作完成以后，点$1$到点$n$的最短距离尽可能大。输出这个最短距离的最大可能值。注意，图中所有边（包括新增边）的边长均为$1$。

输入格式：
第一行包含三个整数$n,m,k$。第二行包含$k$个整数$a_1,a_2,\dots ,a_k$，表示$k$个特殊点的编号，$a_i$之间两两不同。接下来$m$行，每行包含两个整数$x,y$，表示点$x$和点$y$之间存在一条边。

输出格式：
一个整数，表示最短距离的最大可能值。

数据范围：
前六个测试点满足$2\le n\le 100$。
所有测试点满足$2\le n\le 2\times 10^5$，$n-1\le m\le 2\times 10^5$，$2\le k\le n$，$1\le a_i\le n$，$1\le x,y\le n$。

增加完边之后，任选两个特殊点$x$和$y$，所有的最短路可以分为三类（这句话的意思是，最短路有很多条，它们长度一样，但具体的路径可以分为三类），不经过点$x$和$y$的（如果存在的话，长度一定是$1$到$n$的最短路），经过$x\to y$这条边的，和经过$y\to x$这条边的（当然只能经过一次）。设$d_a[c]$表示从$a$到$c$的最短路，那么第一类如果存在的话，值就是$d_1[n]$；第二类就是$d_1[x]+1+d_n[y]$；第三类就是$d_1[y]+1+d_n[x]$。现在要求 max ⁡ x ≠ y { min ⁡ { d 1 [ x ] + 1 + d n [ y ] , d 1 [ y ] + 1 + d n [ x ] , d 1 [ n ] } } = min ⁡ { max ⁡ x ≠ y { min ⁡ { d 1 [ x ] + 1 + d n [ y ] , d 1 [ y ] + 1 + d n [ x ] } , d 1 [ n ] } } \max_{x\ne y}\{\min\{d_1[x]+1+d_n[y], d_1[y]+1+d_n[x],d_1[n]\}\}\\=\min\{\max_{x\ne y}\{\min\{d_1[x]+1+d_n[y], d_1[y]+1+d_n[x]\},d_1[n]\}\} x​=ymax​{min{d1​[x]+1+dn​[y],d1​[y]+1+dn​[x],d1​[n]}}=min{x​=ymax​{min{d1​[x]+1+dn​[y],d1​[y]+1+dn​[x]},d1​[n]}}考虑不等式$d_1[x]+1+d_n[y]\le d_1[y]+1+d_n[x]$，变形$d_1[x]-d_n[x]\le d_1[y]-d_n[y]$，所以可以将$k$个特殊点按照$d_1[x]-d_n[x]$这个权重从小到大排序，这样如果选择了$x<y$这两个点，那么就有 min ⁡ { d 1 [ x ] + 1 + d n [ y ] , d 1 [ y ] + 1 + d n [ x ] } = d 1 [ x ] + 1 + d n [ y ] \min\{d_1[x]+1+d_n[y], d_1[y]+1+d_n[x]\}=d_1[x]+1+d_n[y] min{d1​[x]+1+dn​[y],d1​[y]+1+dn​[x]}=d1​[x]+1+dn​[y]，保持$x<y$并枚举$y$，那么要使得$d_1[x]+1+d_n[y]$尽可能大，也就是要取最大的$d_1[x]$，可以用一个变量存储$<y$的最大$d_1[x]$。$d_1,d_n$这两个数组可以直接用BFS来求。代码如下：

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

const int N = 2e5 + 10, M = 4e5 + 10;
int n, m, k;
int h[N], e[M], ne[M], idx;
int dist1[N], dist2[N], p[N];
int q[N], hh, tt;

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

void bfs(int s, int dist[]) {
  memset(dist, 0x3f, N << 2);
  dist[s] = 0;
  int hh = 0, tt = 0;
  q[tt++] = s;
  while (hh < tt) {
    int t = q[hh++];
    for (int i = h[t]; ~i; i = ne[i]) {
      int v = e[i];
      if (dist[v] > dist[t] + 1) {
        dist[v] = dist[t] + 1;
        q[tt++] = v;
      }
    }
  }
}

int main() {
  scanf("%d%d%d", &n, &m, &k);
  memset(h, -1, sizeof h);
  for (int i = 0; i < k; i++) scanf("%d", &p[i]);
  while (m--) {
    int a, b;
    scanf("%d%d", &a, &b);
    add(a, b), add(b, a);
  }

  bfs(1, dist1), bfs(n, dist2);
  // 将特殊点按照d1[x] - d2[x]来排序
  sort(p, p + k, [&](int x, int y) {
    return dist1[x] - dist2[x] < dist1[y] - dist2[y];
  });

  int res = 0, x = dist1[p[0]];
  for (int i = 1; i < k; i++) {
    int t = p[i];
    res = max(res, dist2[t] + x + 1);
    x = max(x, dist1[t]);
  }

  printf("%d\n", min(res, dist1[n]));
}

时间复杂度$O(n+m+k\log k)$，空间$O(n)$。