loj#P3849. 「NOI2022」树上邻域数点

「NOI2022」树上邻域数点

注意事项

在 LibreOJ 上,由于语言限制,目前只支持以下语言的提交:

  • C++(标准为 C++ 14 及以上)

请在提交源代码前添加 #include "count.h"

题目描述

给出五元组 (T,I,SV,SE,ι)(T, I, S_V, S_E, \iota),其中:

  • TT 是一棵 nn 个点的有根树 T=(V,E)T = (V, E),其中 VVTT 的点集,EETT 的边集。树的节点被编号为 1,2,,n1, 2, \ldots, n,其中根节点编号为 11
  • II 是一个集合,集合中的元素称作信息。其中有两个不同的特殊元素:单位元 ϵ\epsilon 和不合法信息 \bot

对于一般的信息,其都具有点集合边集合两个属性。特别的,对于单位元,其只有边集合的属性,而对于不合法信息,其没有以上两种属性。

  • 对于信息 oI{ϵ,}o \in I \setminus \{ \epsilon, \bot \}oo点集合VV 的一个二元子集,记作 SV(o)S_V(o),满足 SV(o)VS_V(o) \subseteq VSV(o)=2\lvert S_V(o) \rvert = 2。其中,两个集合 A,BA, B 的差 ABA \setminus B 被定义为 $A \setminus B = \{ x \in A \hspace{3mu}\vert\hspace{3mu} x \notin B \}$。
  • 对于信息 oI{}o \in I \setminus \{ \bot \}oo边集合EE 的一个子集,记作 SE(o)S_E(o),满足 SE(o)ES_E(o) \subseteq E。规定单位元的边集合为空,也即 SE(ϵ)=S_E(\epsilon) = \varnothing
  • 对于树上的任何一条边 eEe \in E,记 e=(u,v)e = (u, v),存在一个关于 ee 的信息 ι(e)I\iota(e) \in I,它以其端点为点集合、自身为边集合,即 SV(ι(e))={u,v}S_V(\iota(e)) = \{ u, v \}SE(ι(e))={e}S_E(\iota(e)) = \{ e \}

信息有两种合并的方式,分别记作 RRCC。对于 a,bI\forall a, b \in I,记 r=R(a,b),c=C(a,b)r = R(a, b), c = C(a, b),满足 r,cIr, c \in I,则:

  • 单位元和任何信息合并都得到对方。也即,如果 a=ϵa = \epsilon,那么 r=c=br = c = b;如果 b=ϵb = \epsilon,那么 r=c=ar = c = a
  • 不合法信息和任何信息合并都得到不合法信息。也即,如果 a=a = \bot 或者 b=b = \bot,那么 r=c=r = c = \bot
  • 对于剩下的情况,如果两个信息的边集合的交集非空,或者点集合的交集的大小不为 11,则合并得到不合法信息。也即,如果 SE(a)SE(b)S_E(a) \cap S_E(b) \ne \varnothingSV(a)SV(b)1\lvert S_V(a) \cap S_V(b) \rvert \ne 1,则 r=c=r = c = \bot
  • 否则,有SE(r)=SE(c)=SE(a)SE(b),S_E(r) = S_E(c) = S_E(a) \cup S_E(b) \text{,} SV(r)=SV(a),S_V(r) = S_V(a) \text{,} SV(c)=SV(a)SV(b),S_V(c) = S_V(a) \oplus S_V(b) \text{,} 其中 \oplus 表示集合的对称差运算,也即 AB=(AB)(AB)A \oplus B = (A \cup B) \setminus (A \cap B)

定义 TT 中两个点的树上距离为树上以两个点为端点的唯一简单路径经过的边数。

给出评分参数 MMqq 次询问,每次询问给出树上的一个点 uu 和一个非负整数 dd。记点集 XXTT 中所有与 uu 的树上距离不超过 dd 的点构成的集合,又记边集 $Y = \{ (a, b) \in E \hspace{3mu}\vert\hspace{3mu} a, b \in X \}$ 为 XX 内部的边集。可以证明,从 ϵ\epsilon 和所有 ι(e)\iota(e)eEe \in E)出发,总是能通过有限次 R,CR, C 的调用得到信息 oo \ne \bot 满足 SE(o)=YS_E(o) = Y

每组询问中,你需要在 RRCC 的调用次数总和不超过 MM 的限制下构造出一个满足这样的要求的信息 oo。特别地,如果 d=0d = 0,则直接返回单位元 ϵ\epsilon 即可。

实现细节

请确保你的程序开头有 #include "count.h"

头文件 count.h 中实现了如下内容:

  1. 定义了信息对应的数据类型 info

  2. 定义了 ϵ\epsilon 所对应的 info 类型常量 emptyinfo,你可以在程序中直接使用。

  3. 定义并实现了以下两个信息合并函数,你可以在程序中直接调用:

    info MR(info a,info b);
    info MC(info a,info b);
    
    • 两个函数分别返回 R(a,b)R(a, b)C(a,b)C(a, b) 对应的信息。

    你需要保证调用 R(a,b)\boldsymbol{R(a, b)}C(a,b)\boldsymbol{C(a, b)} 时结果不为 \boldsymbol{\bot},否则程序可能会出现异常行为。

  4. 定义并实现了判定一个信息是否为单位元的函数,你可以在程序中直接调用:

    bool isempty(info a);
    
    • 这个函数返回真当且仅当 aa 为单位元。

可以查看参考交互库了解更多实现细节。

你不需要,也不应该实现主函数。 你需要实现如下几个函数:

void init(int T, int n, int q, vector<int> fa, vector<info> e, int M);
  • T 表示测试点编号,n 表示树的点数,q 表示询问数,M 表示该测试点的评分参数。
  • fae 的长度均为 n1n - 1。对于 0i<n10 \le i < n - 1fa[i]fa[i]i+2i + 2 为第 ii 条边 eie_i 的两个端点,e[i]e[i] 为题目描述中提到的 ι(ei)\iota(e_i) 所对应的 info 类型元素。数据保证 fa[i]fa[i] 小于 i+2i + 2
info ask(int u, int d);

给出一个询问,参数的意义见题目描述。你需要在函数结束时返回一个满足题设条件的信息。

最终测试时,在每个测试点,交互库会恰好调用一次 init 函数,随后调用 qqask函数。交互库会使用特殊的实现方式,单个 info 类型的变量会恒定消耗 1212 字节内存,这与下发的参考交互库不同。为保证程序运行时内存使用在题目限制内,你需要保证运行过程中没有过多的 info 类型变量同时存在。

保证在满足调用次数限制且不进行 isempty 函数调用的情况下,最终测试的交互库运行所需的时间不超过 0.6 秒,交互库本身所消耗的内存不超过 16 MiB。保证在只执行 108{10}^8isempty 函数调用的情况下,最终测试的交互库运行的时间不超过 0.25 秒。

在下发文件中包含一个名为 count.cpp 的文件,作为示例程序,选手可以在此基础上继续实现本题。在下发文件中还额外包含一个名为 count_backup.h 的备份文件,我们保证其与 count.h 文件完全相同。

测试程序方式

本题目录下提供了两个交互库的参考实现 grader.ochecker.o,其为两个不同的交互库编译产生的可链接文件。最终测试时所用的交互库实现与该实现有不同,因此选手的解法不应依赖交互库的具体实现,同时也不应该依赖 count.hinfo 类型的具体实现。

你需要修改下发的 count.h 来帮助进行链接。具体的,在将源代码 count.cpp 和程序 grader.o 进行链接的时候,你需要注释掉 count.h 代码的第 5 行,并保留第 4 行的代码。链接 checker.o 方法类似,需要注释掉 count.h 代码的第 4 行,并保留第 5 行的代码。选手可以对 count.h 的实现自行修改来实现不同程序的编译。

修改后,选手可以在本题目录下使用如下命令编译得到可执行程序:

g++ count.cpp -c -O2 -std=c++14 -lm && g++ count.o grader.o -o count
g++ count.cpp -c -O2 -std=c++14 -lm && g++ count.o checker.o -o count

其中第一行命令会编译当前 count.cpp 后与 grader.o 链接起来,生成可执行文件 count,第二行命令则会编译当前 count.cpp 后与 checker.o 链接起来,生成可执行文件 count

按上述方法编译得到的可执行文件 count,其运行方式如下:

  • 可执行文件将从标准输入读入以下格式的数据:
    • 第一行四个整数 id,n,q,Mid, n, q, M,分别表示测试点编号、树的点数、询问数和评分参数;
    • 第二行 n1n - 1 个整数 p2,p3,,pnp_2, p_3, \ldots, p_n,分别表示 22nn 的父亲节点编号,在本地调试时你需要保证 i[2,n]\forall i \in [2, n]pi<ip_i < i
    • 接下来 qq 行每行两个整数 u,du, d,描述一次询问。
  • 读入之后,交互库会进行测试。如果你的程序不满足交互库限制,其会在输出中返回对应的错误信息。否则,对于链接的可执行文件,其输出如下:
    • 总共一行三个整数 C1,C2,C3C_1, C_2, C_3,其中:
      • C1C_1 表示程序在 init 函数中调用交互库函数的总次数;
      • C2C_2 表示程序在运行过程中调用交互库函数的总次数;
      • C3C_3 表示程序在 qqask 函数中调用交互库函数的次数的最大值。
      • 对于上述三个统计量,我们只会计入 MRMC 函数的调用次数,而不会计入 isempty 函数的调用次数。
  • 在链接不同文件的时候,其能够进行的检查也不同,具体地:
    • grader.o:其在运行时不会检查 ask 函数返回的信息是否正确,但可以帮助选手判断交互操作是否符合要求。这份程序运行时间最接近评测时的交互库,因此选手可以利用该程序测试运行速度,但不保证程序正确性。
    • checker.o:其在运行时会检查 ask 函数返回的信息是否正确,也可以帮助选手判断交互操作是否符合要求。同时其会检查 ask 函数返回的信息是否正确。这份程序可以进行答案正确性的检查。

选手在调试时需要保证输入可执行文件 count 的数据满足上述输入格式,否则不保证输出结果正确。

样例 1

见附件中的 count/count1.incount/count1.ans

样例 2

见附件中的 count/count2.incount/count2.ans

该组样例满足数据范围中的特殊性质 A。

样例 3

见附件中的 count/count3.incount/count3.ans

该组样例满足数据范围中的特殊性质 B。

样例 4

见附件中的 count/count4.incount/count4.ans

评分方式

最终评测只会收取 count.cpp,修改选手目录下其他文件不会对评测结果产生影响。注意:

  • 未初始化的 info 类型的变量不保证是 emptyinfo
  • 请不要尝试访问或修改 info 类型的成员变量,否则将被视为攻击交互库。
  • 请不要在 init 函数调用之前调用 MRMC 函数,否则可能会发生未定义行为。
  • 你只能访问自己定义的变量和交互库返回的 info 类型变量,尝试访问其他空间将可能导致编译错误或运行时错误。

本题首先会受到和传统题相同的限制,例如编译错误会导致整道题目得 0 分,运行时错误、超过时间限制、超过空间限制等会导致相应测试点得 0 分等。

在上述条件以外,在一个测试点中,若程序执行了非法的函数调用或询问操作中给出了错误回答,该测试点将会获得 0 分。否则,记 C1,C3C_1, C_3 分别表示你的程序在 init 函数中调用交互库函数的次数,和你的程序在所有 qqask 函数中调用交互库函数的次数的最大值。如果 C13107C_1 \le 3 \cdot {10}^7C3C_3 不超过该测试点的评分参数 MM,你将获得该测试点的分数,否则你无法获得该测试点的分数。注意:计算 C1,C3C_1, C_3 时只会计入 MRMC 函数的调用次数,而不会计入 isempty 函数的调用次数。

数据范围与提示

对于所有测试点,1n2×1051 \le n \le 2 \times {10}^51q1061 \le q \le {10}^6;每组询问中,有 1un1 \le u \le n1dn11 \le d \le n - 1

测试点 n=n= q=q= 特殊性质 M=M=
11 10001000 10410^4 500500
22 20002000
3,43,4 10510^5 10610^6 A 55
5,65,6 6×1046 \times 10^4 6×1046\times 10^4 B 5050
77 6×1046 \times 10^4 55
88 10510^5
99 75007500 5×1045 \times 10^4 C 500500
1010 10410^4
1111 1.5×1041.5 \times 10^4
1212 2×1042 \times 10^4 5050
1313 2.5×1042.5 \times 10^4 55
1414 3×1043 \times 10^4 10510^5
1515 6×1046 \times 10^4 10610^6 D
1616
1717 8×1048 \times 10^4
1818 10510^5
1919 1.5×1051.5 \times 10^5
2020 2×1052 \times 10^5 11

特殊性质 A:保证 i[1,n1]\forall i \in [1, n - 1],编号为 i+1i + 1 的点的父节点为 ii
特殊性质 B:保证所有询问均满足 u=1u = 1
特殊性质 C:保证所有询问均满足 d100d \le 100
特殊性质 D:保证所有询问均满足 d1000d \ge 1000