## [$AcWing$ $1164$. 加工零件](https://www.acwing.com/problem/content/description/1166/)
### 一、题目描述
凯凯的工厂正在有条不紊地生产一种神奇的零件,神奇的零件的生产过程自然也很神奇。
工厂里有 $n$ 位工人,工人们从 $1∼n$ 编号。
某些工人之间存在 **双向** 的零件传送带。
保证每两名工人之间最多只存在一条传送带。
如果 $x$ 号工人想生产一个被加工到第 $L(L>1)$ 阶段的零件,则所有与 $x$ 号工人有传送带直接相连的工人,都需要生产一个被加工到第 $L−1$ 阶段的零件(但 $x$ 号工人自己无需生产第 $L−1$ 阶段的零件)。
如果 $x$ 号工人想生产一个被加工到第 $1$ 阶段的零件,则所有与 $x$ 号工人有传送带直接相连的工人,都需要为 $x$ 号工人提供一个原材料。
轩轩是 $1$ 号工人。
现在给出 $q$ 张工单,第 $i$ 张工单表示编号为 $a_i$ 的工人想生产一个第 $L_i$ 阶段的零件。
轩轩想知道 **对于每张工单,他是否需要给别人提供原材料**。
他知道聪明的你一定可以帮他计算出来!
**输入格式**
第一行三个正整数 $n$,$m$ 和 $q$,分别表示工人的数目、传送带的数目和工单的数目。
接下来 $m$ 行,每行两个正整数 $u$ 和 $v$,表示编号为 $u$ 和 $v$ 的工人之间存在一条零件传输带。保证 $u≠v$。
接下来 $q$ 行,每行两个正整数 $a$ 和 $L$,表示编号为 $a$ 的工人想生产一个第 $L$ 阶段的零件。
**输出格式**
共 $q$ 行,每行一个字符串 “$Yes$” 或者 “$No$”。如果按照第 $i$ 张工单生产,需要编号为 $1$ 的轩轩提供原材料,则在第 $i$ 行输出 “$Yes$”;否则在第 $i$ 行输出 “$No$”。注意输出不含引号。
**数据范围**
$1≤u,v,a≤n$。
测试点 $1∼4$,$1≤n,m≤1000$,$q=3$,$L=1$。
测试点 $5∼8$,$1≤n,m≤1000$,$q=3$,$1≤L≤10$。
测试点 $9∼12,1≤n,m,L≤1000,1≤q≤100$。
测试点 $13∼16,1≤n,m,L≤1000,1≤q≤105$。
测试点 $17∼20,1≤n,m,q≤105,1≤L≤109$。
**输入样例1**:
```cpp {.line-numbers}
3 2 6
1 2
2 3
1 1
2 1
3 1
1 2
2 2
3 2
```
**输出样例1**:
```cpp {.line-numbers}
No
Yes
No
Yes
No
Yes
```
**输入样例2**:
```cpp {.line-numbers}
5 5 5
1 2
2 3
3 4
4 5
1 5
1 1
1 2
1 3
1 4
1 5
```
**输出样例2**:
```cpp {.line-numbers}
No
Yes
No
Yes
Yes
```
### 二、抽象题意
给定一张包含 $n$ 个点和 $m$ 条边的 **无向图**,再给定 $q$ 个询问:$a_i,L_i$,判断是否存在一条从$1$号点走到$a_i$号点的 **恰好** 经过$L$条边的路径。
设求第$3$点为第$3$阶段时,点$1$是否需要提供原材料。
$【3,3】 => 【2,2】,【4,2】$
$【2,2】 => 【1,1】,【3,1】$
$【4,2】 => 【3,1】,【5,1】$
$【1,1】 => 【2,0】,【5,0】$
$【3,1】 => 【2,0】,【4,0】$
$【5,1】 => 【1,0】,【4,0】$ # 此处$1$需要提供原材料
比较容易想到对于每个询问进行暴搜,若点$1$为$0$,则$Yes$, 时间复杂度很高,必然超时。
```cpp {.line-numbers}
#include
using namespace std;
typedef pair PII;
#define x first
#define y second
const int N = 1e5 + 10, M = N << 1;
// 20个测试点,过了7个,得了35分,其它TLE
// 邻接表
int e[M], h[N], idx, ne[M];
void add(int a, int b) {
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
bool bfs(int start, int dist) {
queue q;
q.push({start, dist});
while (q.size()) {
auto u = q.front();
q.pop();
if (u.x == 1 && u.y == 0) return true; // 在用完所有步数后,到达1号点,成功!
for (int i = h[u.x]; ~i; i = ne[i]) {
int v = e[i];
if (u.y) q.push({v, u.y - 1});
}
}
return false;
}
int main() {
memset(h, -1, sizeof h);
int n, m, Q;
cin >> n >> m >> Q;
while (m--) {
int a, b;
cin >> a >> b;
add(a, b), add(b, a); // 无向图,双向建边
}
while (Q--) {
int u, dist;
cin >> u >> dist;
cout << (bfs(u, dist) ? "Yes" : "No") << endl;
}
return 0;
}
```
### 四、优化算法
(最短路,$BFS$) $O(n+m+q)$
**本题关键点**:
> 如果存在一条长度是 $L$ 的路径,其中 $L>0$,那么我们可以在其中任意一条边上来回走,就可以构造出来长度是 $L+2,L+4,L+6,…$的路径。
因此当我们想判断是否存在长度是 $L$ 的路径时,只需判断是否 **存在长度小于等于 $L$**,且 **奇偶性** 和 $L$ 相同的路径即可。
因此我们可以预处理出从$1$号点出发,到每个点长度为 **奇数的最短路径** 和 **长度为偶数的最短路径**。
这里可以使用拆点技巧来构造新图,类似于$DP$中的[状态机模型](https://www.acwing.com/problem/content/1051/):
* 将原图中的每个点 $v$ 拆成两个新点:偶点$v_0$和奇点$v_1$;
* 将原图中的每条边 $(u,v)$ 拆成两条新边:$(u_0,v_1)$ 和 $(u_1,v_0)$;
那么在新图中从 $1$ 号点走到 $v_0$ 的所有路径,对应在原图中从 $1$ 号点走到 $v$ 的所有 **长度是偶数** 的路径;在新图中从 $1$ 号点走到 $v_1$ 的所有路径,对应在原图中从 $1$ 号点走到 $v$ 的所有 **长度是奇数** 的路径。
因此在新图上求出$1$号点到其他所有点的最短路径,即可求出在原图中从$1$号点到其他所有点的长度是奇数和偶数的最短路径。
由于所有边的长度为$1$,因此可以用$BFS$求最短路。
以上我们处理了 $L_i>0$ 的情况,还需特判一下 $L_i==0$ 的情况,$L_i==0$ 表示一条边都不存在,即$1$号点与其他点均不连通,此时由于输入时 $L_i≥1$,因此直接输出 $No$即可。
#### 时间复杂度
使用$BFS$求最短路的时间复杂度是 $(n+m)$;
判断每个查询操作的时间复杂度是 $O(1)$;
因此总时间复杂度是 $O(n+m+q)$。
```cpp {.line-numbers}
#include
using namespace std;
typedef pair PII;
#define x first // 节点号
#define y second // 奇偶性
const int N = 100010, M = 200010;
int n, m, Q;
int h[N], e[M], ne[M], idx;
void add(int a, int b) {
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
int dist[N][2]; // 奇偶最短路
queue q;
// 因边边权都是1,不需要Dijkstra或者Floyd,只需要bfs即可
// 由于本题是强调 奇偶性 ,所以,需要使用 拆点 的思想
// 所谓拆点,就是像状态机或者DP中扩展维度的思想,将奇偶点分开,把混沌的事情清晰化。
// dist[i][0]:表示是偶数步的点,值是最小步数
// dist[i][1]:表示是奇数步的点
void bfs() {
memset(dist, 0x3f, sizeof dist); // 预求最短,先设最大
dist[1][0] = 0; // 1是起点,走0步到达即偶点,步数:0
q.push({1, 0}); // 起点入队列
while (q.size()) {
auto u = q.front();
q.pop();
for (int i = h[u.x]; ~i; i = ne[i]) { // u点
PII v = {e[i], u.y ^ 1};
// u.y ^ 1 奇<->偶 互换
// 由于上面的 y=u.y ^ 1,使得(u.x,u.y)只向(x,u.y的奇偶变换后数) 转移
// 这样做,就起到的拆点作用,而并不是真正的把点拆开
if (dist[v.x][v.y] > dist[u.x][u.y] + 1) { // 如果利用u可以更新v这个点
dist[v.x][v.y] = dist[u.x][u.y] + 1; // 更新
q.push(v); // v的最短距离更新了,入队列,去更新更多的点
}
}
}
}
int main() {
cin >> n >> m >> Q;
memset(h, -1, sizeof h);
while (m--) {
int a, b;
cin >> a >> b;
add(a, b), add(b, a);
}
bfs();
while (Q--) {
int a, l;
cin >> a >> l;
/*
如果不写下面的特判,会被欺负死:
3 1 1
2 3
1 2
第一行三个正整数 n,m 和 q,分别表示工人的数目、传送带的数目和工单的数目。
也就是3个工人,1个传送带,1个工单
传送带是2~3,也就是1是与世隔绝的
接下来 q 行,每行两个正整数 a 和 L,表示编号为 a 的工人想生产一个第 L 阶段的零件。
1号工人,想生产2阶段的零件
由于1号工人是与世隔绝的,没有人给它提供1阶段的零件,肯定返回No,此时需要特判,返回答案
*/
if (h[1] == -1)
puts("No");
else if (l >= dist[a][l & 1]) // l是奇数,并且大于奇数最短路,则可以通过+2,+4,+6...等方法来回磨,凑够l就完事
// l是偶数,并且大于偶数最短路,则可以通过+2,+4,+6...等方法来回磨,凑够l就完事
puts("Yes");
else
puts("No");
}
return 0;
}
```