[CSP-J 2019] 零件加工

问题描述

凯凯的工厂正在有条不紊地生产一种神奇的零件，神奇的零件的生产过程自然也很神奇。工厂里有 n 位工人，工人们从 1 ~ n 编号，某些工人之间存在双向的零件传送带。保证每两名工人之间最多只存在一条传送带。

如果 x 号工人想生产一个被加工到第 L(L≥1) 阶段的零件，则 所有 与 x 号工人有传送带 直接 相连的工人，都需要生产一个被加工到第 L−1 阶段的零件（但 x 号工人自己 无需 生产第 L−1 阶段的零件）。

如果 x 号工人想生产一个被加工到第 1 阶段的零件，则 所有 与 x 号工人有传送带 直接 相连的工人，都需要为 x 号工人提供一个原材料。

轩轩是 1 号工人。现在给出 q 张工单，第 i 张工单表示编号为 ai​ 的工人想生产一个第 Li​ 阶段的零件。轩轩想知道对于每张工单，他是否需要给别人提供原材料。他知道聪明的你一定可以帮他计算出来！

输入格式

第一行三个正整数 n,m 和 q，分别表示工人的数目、传送带的数目和工单的数目。

接下来 m 行，每行两个正整数 u 和 v，表示编号为 u 和 v 的工人之间存在一条零件传送带。保证 u≠v。

接下来 q 行，每行两个正整数 a 和 L，表示编号为 a 的工人想生产一个第 L 阶段的零件。

输出格式

共 q 行，每行一个字符串“Yes”或者“No”。如果按照第 i 张工单生产，需要编号为 11 的轩轩提供原材料，则在第 i 行输出“Yes”；否则在第 i 行输出“No”。注意输出不含引号。

样例 1 输入

3 2 6
1 2
2 3
1 1
2 1
3 1
1 2
2 2
3 2

样例 1 输出 

No
Yes
No
Yes
No
Yes

样例 1 解释 

编号为 1 的工人想生产第 1 阶段的零件，需要编号为 2 的工人提供原材料。

编号为 2 的工人想生产第 1 阶段的零件，需要编号为 1 和 3 的工人提供原材料。

编号为 3 的工人想生产第 1 阶段的零件，需要编号为 2 的工人提供原材料。

编号为 1 的工人想生产第 2 阶段的零件，需要编号为 2 的工人生产第 1 阶段的零件，需要为编号 1 和 3 的工人提供原材料。

编号为 2 的工人想生产第 2 阶段的零件，需要编号为 1 和 3 的工人生产第 1 阶段的零件，他/她们都需要编号为 2 的工人提供原材料。

编号为 3 的工人想生产第 2 阶段的零件，需要编号为 2 的工人生产第 1 阶段的零件，需要编号为 1 和 3 的工人提供原材料。

其实本题是可以通过预处理进行优化。就是先不管询问，直接从 1 出发走 1000 步，把搜索过程中遇到的所有状态使用 vis[][] 进行标记。

这样对于每一次询问，直接根据之前记录好的 vis[a][L] 情况，判断合法输出即可。

时间复杂度 O(mL) ，可以通过前 16 个测试点

代码如下：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
struct node {
    int x;
    int step;
    node() {
        
    }
    node(int _x, int _step) {
        x = _x;
        step = _step;
    }
};
vector < int > G[1005];
int vis[1005][1005];
queue < node > q;
void bfs(int s) {
    vis[s][0] = true;
    q.push(node(s,0));
    while (!q.empty()) {
        node now = q.front();
        q.pop();
        if (now.step > 1000){
            continue;
        }
        for (int i = 0; i < G[now.x].size(); i++) {
            int v = G[now.x][i];
            if (!vis[v][now.step+1]){
                vis[v][now.step+1] = 1;
                q.push(node(v,now.step+1));
            }
        }
    }
}
int main() {
    int n, m, q, a, L;
    scanf("%d%d%d", & n, & m, & q);
    for (int i = 0; i < m; i++) {
        int u, v;
        scanf("%d%d", & u, & v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    bfs(1);
    while (q--) {
        scanf("%d%d", & a, & L);
        if (vis[a][L]){
            printf("Yes\n");
        } else {
            printf("No\n");
        }
    }
    return 0;
}