#P8519. [IOI2021] 钥匙

[IOI2021] 钥匙

题目背景

滥用本题评测将被封号

由于技术限制,请不要使用 C++ 14 (GCC 9) 提交本题。

这是一道交互题,你只需要实现代码中要求的函数。

你的代码不需要引用任何额外的头文件,也不需要实现 main 函数。

题目描述

建筑师 Timothy 设计了一个新的密室逃脱游戏。

这个游戏里有 nn 个房间,房间从 00n1n - 1 编号。最开始的时候,每个房间里都恰好有一把钥匙。每把钥匙都有一个类型,钥匙的类型是一个 00n1n - 1 区间内的整数。第 ii 个房间里的钥匙类型是 r[i]r[i]。注意多个房间里可能会包含相同类型的钥匙,即r[i]r[i] 的值不一定是两两不同的。

游戏里还有 mm双向的通道,通道从 00m1m - 1 编号。第 jj 条通道连接了一对编号不同的房间 u[j]u[j]v[j]v[j]。同一对房间之间可能存在多条通道。

参与游戏的玩家需要收集钥匙和在不同的房间之间通过通道进行移动。 当玩家使用通道 jj 从房间 u[j]u[j] 移动到 v[j]v[j] ,或者反过来从 v[j]v[j] 移动到 u[j]u[j] 时,我们说玩家通过了通道 jj。 只有当玩家收集到类型为 c[j]c[j] 的钥匙时,玩家才可以通过通道 jj

在游戏的任意时刻,玩家可以在某个房间 xx 里执行以下两种操作:

  • 收集房间 xx 里面的钥匙,钥匙的类型是 r[x]r[x](除⾮对应类型的钥匙已经被收集过)。
  • 通过通道 jj,需要满足 u[j]=xu[j] = xv[j]=xv[j] = x,且玩家已经获得 c[j]c[j] 类型的钥匙。

注意玩家收集过的钥匙可以一直使用,永远不会被丢弃

最初玩家会在某个房间 ss 开始游戏,不带任何钥匙。 如果玩家从房间 ss 开始,通过一系列上述描述的两种操作,能够到达房间 tt,那么称房间 tt从房间 ss 开始可以到达的

对于每一个房间 i (0in1)i ~ ( 0 \le i \le n − 1),定义从房间 ii 出发能够到达的房间数为 p[i]p[i]。Timothy 想要知道满⾜ p[i]p[i] 值最小的下标 ii 的集合。

输入格式

你要实现以下函数:

std::vector<int> find_reachable(
	std::vector<int> r, std::vector<int> u, 
	std::vector<int> v, std::vector<int> c
)
  • rr:⼀个⻓度为 nn 的序列。对于每⼀个 i (0in1)i ~ ( 0 \leq i \leq n − 1),第 ii 个房间⾥的钥匙类型是 r[i]r[i]
  • u,vu, v:两个⻓度为 mm 的序列。 对于每⼀个 j (0jm1)j ~ ( 0 \leq j \leq m − 1),第 jj 条通道连接了房间 u[j]u[j]v[j]v[j]
  • cc:⼀个⻓度为 mm 的序列。对于每⼀个 j (0jm1)j ~ ( 0 \leq j \leq m − 1),通过通道 jj 需要⽤到的钥匙类型是 c[j]c[j].

输出格式

  • 这个函数应该返回⼀个⻓度为 nn 的序列 aa。对于 0in10 \leq i \leq n − 1 中的 ii,如果满⾜ p[i]p[j] (0jn1)p[i] \leq p[j] ~ (0 \leq j \leq n − 1) 那么 a[i]a[i] 的值为 11,否则 a[i]a[i] 的值为 00
4 5
0 1 1 2
0 1 0
0 2 0
1 2 1
1 3 0
3 1 2

0 1 1 0

7 10
0 1 1 2 2 1 2
0 1 0
0 2 0
1 2 1
1 3 0
2 3 0
3 4 1
3 5 2
4 5 0
4 6 2
5 6 1

0 1 1 0 1 0 1

3 1
0 0 0
0 1 0

0 0 1

提示

样例解释

对于例 11,考虑以下调用:

find_reachable([0, 1, 1, 2],
[0, 0, 1, 1, 3], [1, 2, 2, 3, 1], [0, 0, 1, 0, 2])

如果玩家从房间 00 开始游戏,可以执⾏以下的操作序列:

当前房间 操作
00 收集钥匙类型 00
通过通道 00 到房间 11
11 收集钥匙类型 11
通过通道 22 到房间 22
22 通过通道 22 到房间 11
11 通过通道 33 到房间 33

因此从房间 00 出发可以到达房间 33。 类似地,我们可以构造出操作序列表明所有 44 个房间都是从房间 00 出发可达的,所以 p[0]=4p[0] = 4。 下表展⽰了从各个房间出发可以到达的房间集合:

开始房间 ii 可以到达的房间 p[i]p[i]
00 [0,1,2,3][0, 1, 2, 3] 44
11 [1,2][1, 2] 22
22
33 [1,2,3][1, 2, 3] 33

所有房间中 p[i]p[i] 的最小值是 22,这可以在 i=1i = 1i=2i = 2 处取得。所以这次函数调⽤的返回值是 [0,1,1,0][0, 1, 1, 0]

对于例 22:所有房间中 p[i]p[i] 的最小值是 22,这可以在 i{1,2,4,6}i \in \{1, 2, 4, 6\} 处取得。所以这次函数调⽤的返回值是 [0,1,1,0,1,0,1][0, 1, 1, 0, 1, 0, 1]

对于例 33:所有房间中 p[i]p[i] 的最小值是 11,这可以在 i=2i = 2 处取得。所以这次函数调⽤的返回值是 [0,0,1][0, 0, 1]

约束条件

  • 2n3×1052 \leq n \leq 3 \times 10 ^ 5
  • 1m3×1051 \leq m \leq 3 \times 10 ^ 5
  • 0r[i]n10 \leq r[i] \leq n - 1 (对于所有的 0in10 \leq i \leq n - 1
  • 0u[j],v[j]n10 \leq u[j], v[j] \leq n - 1u[j]v[j]u[j] \neq v[j] (对于所有的 0jm10 \leq j \leq m - 1
  • 0c[j]n10 \leq c[j] \leq n - 1(对于所有的 0jm10 \leq j \leq m - 1

子任务

  1. 99 分) c[j]=0c[j] = 0(对于所有的 0jm10 \leq j \leq m − 1)且 n,m200n, m \leq 200
  2. 1111 分) n,m20n, m \leq 20
  3. 1717 分) n,m2000n, m \leq 2000
  4. 3030 分) c[j]29c[j] \leq 29(对于所有的 0jm10 \leq j \leq m − 1)且 r[i]29r[i] \leq 29(对于所有的 0in10 \leq i \leq n − 1
  5. 3333 分)没有额外的约束条件。

样例评分程序

评测程序⽰例以如下格式读取输⼊数据:

  • 11 ⾏:n mn ~ m
  • 22 ⾏: r[0] r[1]  r[n1]r[0] ~ r[1] ~ \cdots ~ r[n − 1]
  • 3+j3 + j(0jm1)( 0 \leq j \leq m − 1)u[j] v[j] c[j]u[j] ~ v[j] ~ c[j]

样例评分程序按照以下格式打印 find_reachable 函数的返回值:

11 ⾏: s[0] s[1] s[n1]s[0] ~ s[1] \cdots ~ s[n − 1]