loj#P3528. 「IOI2021」位移寄存器

「IOI2021」位移寄存器

题目描述

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

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

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

工程师 Christopher 在开发一款新的计算机处理器。

这个处理器可以访问 mm 个不同的 bb 位存储单元(本题中 m=100m = 100b=2000b = 2000)。它们被称作寄存器,编号从 00m1m - 1。我们把这些寄存器记为 r[0],r[1],,r[m1]r[0], r[1], \ldots , r[m - 1]。每个寄存器都是 bb 个比特的数组,这些比特从 00(最右的比特)到 b1b - 1(最左的比特)编号。对所有的 ii0im10 \le i \le m - 1)和 jj0jb10 \le j \le b - 1),我们将寄存器 ii 的第 jj 个比特记为 r[i][j]r[i][j]

对所有的比特序列 d0,d1,,dl1d_0, d_1, \ldots , d_{l - 1}(具有某个长度 ll),该序列的整数值等于 $2^0 \cdot d_0 + 2^1 \cdot d_1 + \cdots + 2^{l - 1} \cdot d_{l - 1}$。我们说存储在某个寄存器中的整数值就是寄存器中比特序列的整数值,也就是说,该整数值为 $2^0 \cdot r[i][0] + 2^1 \cdot r[i][1] + \cdots + 2^{b - 1} \cdot r[i][b - 1]$。

该处理器有 99 种类型的指令,可以用来修改寄存器中的比特。每条指令操作一个或多个寄存器,并将其输出存储到其中的一个寄存器。下面我们用 x:=yx := y 表示一个修改 xx 的值并将其变成 yy 的操作。每种类型的指令所做的操作描述如下:

  • move(t,y)\operatorname{\mathit{move}}(t, y):将寄存器 yy 中的比特数组拷贝到寄存器 tt。对所有的 jj0jb10 \le j \le b - 1),设置 r[t][j]:=r[y][j]r[t][j] := r[y][j]

  • store(t,v)\operatorname{\mathit{store}}(t, v):设置寄存器 tt 等于 vv,这里 vv 是某个 bb 个比特的数组。对于所有的 jj0jb10 \le j \le b - 1),设置 r[t][j]:=v[j]r[t][j] := v[j]

  • and(t,x,y)\operatorname{\mathit{and}}(t, x, y):取寄存器 xxyy 的按位与,并将结果存到寄存器 tt 中。对于所有的 jj0jb10 \le j \le b - 1),如果 r[x][j]r[x][j]r[y][j]r[y][j] 同时11 则设置 r[t][j]:=1r[t][j] := 1,否则设置 r[t][j]:=0r[t][j] := 0

  • or(t,x,y)\operatorname{\mathit{or}}(t, x, y):取寄存器 xxyy 的按位或,并将结果存到寄存器 tt 中。对于所有的 jj0jb10 \le j \le b - 1),如果 r[x][j]r[x][j]r[y][j]r[y][j] 至少有一个11 则设置 r[t][j]:=1r[t][j] := 1,否则设置 r[t][j]:=0r[t][j] := 0

  • xor(t,x,y)\operatorname{\mathit{xor}}(t, x, y):取寄存器 xxyy 的按位异或,并将结果存到寄存器 tt 中。对于所有的 jj0jb10 \le j \le b - 1),如果 r[x][j]r[x][j]r[y][j]r[y][j] 恰好有一个11 则设置 r[t][j]:=1r[t][j] := 1,否则设置 r[t][j]:=0r[t][j] := 0

  • not(t,x)\operatorname{\mathit{not}}(t, x):取寄存器 xx 的按位非,并将结果存到寄存器 tt 中。对于所有的 jj0jb10 \le j \le b - 1),设置 r[t][j]:=1r[x][j]r[t][j] := 1 - r[x][j]

  • left(t,x,p)\operatorname{\mathit{left}}(t, x, p):左移寄存器 xx 中的所有比特 pp 位,并将结果存到寄存器 tt 中。将寄存器 xx 中的比特左移 pp 位的结果,是一个包含 bb 个比特的数组 vv。对于所有的 jj0jb10 \le j \le b - 1),如果 jpj \ge pv[j]=r[x][jp]v[j] = r[x][j - p],否则 v[j]=0v[j] = 0。对所有的 jj0jb10 \le j \le b - 1),设置 r[t][j]:=v[j]r[t][j] := v[j]

  • right(t,x,p)\operatorname{\mathit{right}}(t, x, p):右移寄存器 xx 中的所有比特 pp 位,并将结果存到寄存器 tt 中。将寄存器 xx 中的比特右移 pp 位的结果,是一个包含 bb 个比特的数组 vv。对于所有的 jj0jb10 \le j \le b - 1),如果 jb1pj \le b - 1 - pv[j]=r[x][j+p]v[j] = r[x][j + p],否则 v[j]=0v[j] = 0。对所有的 jj0jb10 \le j \le b - 1),设置 r[t][j]:=v[j]r[t][j] := v[j]

  • add(t,x,y)\operatorname{\mathit{add}}(t, x, y):将寄存器 xxyy 中的整数值加起来,并将结果存到寄存器 tt 中。加法是在模 2b2^b 下做的。正式一些来说,设 XX 是操作前存在寄存器 xx 中的整数值,而 YY 是操作前存在寄存器 yy 中的整数值。设 TT 为操作后存在寄存器 tt 中的整数值。如果 X+Y<2bX + Y < 2^b,设置 tt 中的比特使得 T=X+YT = X + Y。否则,设置 tt 中的比特使得 T=X+Y2bT = X + Y - 2^b

Christopher 希望你用这个新处理器解决两种任务。任务的类型用整数 ss 来表示。对所有类型的任务,你需要创建一个程序,其为上文所定义的指令构成的序列。

程序的输入包括 nn 个整数 a[0],a[1],,a[n1]a[0], a[1], \ldots , a[n - 1],而每个整数都有 kk 个比特,也就是说,a[i]<2ka[i] < 2^k0in10 \le i \le n - 1)。在程序执行前,输入的所有的数都依次存储在寄存器 00 中,使得对所有的 ii0in10 \le i \le n - 1),kk 比特序列 $r[0][i \cdot k], r[0][i \cdot k + 1], \ldots , r[0][(i + 1) \cdot k - 1]$ 的整数值等于 a[i]a[i]。注意 nkbn \cdot k \le b。寄存器 00 中所有其他的比特(其下标在 nkn \cdot kb1b - 1 之间,包括 nkn \cdot kb1b - 1),以及其他所有寄存器中的所有比特,都初始化为 00

执行某个程序就是按序执行其所包含的指令。在最后一条指令执行完毕后,程序的输出将根据寄存器 00 中比特最终的值计算出来。具体来说,输出是 nn 个整数 c[0],c[1],,c[n1]c[0], c[1], \ldots , c[n - 1] 的序列,这里对所有 ii0in10 \le i \le n - 1)来说,c[i]c[i] 都是寄存器 00 中比特 iki \cdot k(i+1)k1(i + 1) \cdot k - 1 所构成的序列的整数值。注意,在程序运行结束后,寄存器 00 中其余的比特(下标不小于 nkn \cdot k),以及其他寄存器中的所有比特,可能是任意值。

  • 第一个任务(s=0s = 0)是要找出输入整数 a[0],a[1],,a[n1]a[0], a[1], \ldots , a[n - 1] 中的最小值。 具体来说,c[0]c[0] 必须是 a[0],a[1],,a[n1]a[0], a[1], \ldots , a[n - 1] 中的最小值。c[1],c[2],,c[n1]c[1], c[2], \ldots , c[n - 1] 的值可以是任意的。

  • 第二个任务(s=1s = 1)是要将输入整数 a[0],a[1],,a[n1]a[0], a[1], \ldots , a[n - 1] 进行非降序排序。具体来说,对于所有的 ii0in10 \le i \le n - 1),c[i]c[i] 应当等于 a[0],a[1],,a[n1]a[0], a[1], \ldots , a[n - 1] 中第 1+i1 + i 小的整数(也就是说,c[0]c[0] 是输入整数中的最小整数)。

请帮 Christopher 写一下解决这些任务的程序。每个程序至多只能包含 qq 条指令。

实现细节

你要实现如下函数:

void construct_instructions(int s, int n, int k, int q)
  • ss:任务类型。
  • nn:输入中的整数的数量。
  • kk:输入中的每个整数的比特数。
  • qq:允许的最大的指令数。
  • 该函数将被恰好调用一次,并应当为所要解决的任务创建一个指令序列。

该函数应当调用以下函数中的一或多个,以创建指令序列:

void append_move(int t, int y)
void append_store(int t, bool[] v)
void append_and(int t, int x, int y)
void append_or(int t, int x, int y)
void append_xor(int t, int x, int y)
void append_not(int t, int x)
void append_left(int t, int x, int p)
void append_right(int t, int x, int p)
void append_add(int t, int x, int y)
  • 每个函数分别往程序追加一条 move(t,y)\operatorname{\mathit{move}}(t, y)store(t,v)\operatorname{\mathit{store}}(t, v)and(t,x,y)\operatorname{\mathit{and}}(t, x, y)or(t,x,y)\operatorname{\mathit{or}}(t, x, y)xor(t,x,y)\operatorname{\mathit{xor}}(t, x, y)not(t,x)\operatorname{\mathit{not}}(t, x)left(t,x,p)\operatorname{\mathit{left}}(t, x, p)right(t,x,p)\operatorname{\mathit{right}}(t, x, p)add(t,x,y)\operatorname{\mathit{add}}(t, x, y) 指令。
  • 对于所有相关的指令,ttxxyy 必须至少为 00 且至多为 m1m - 1
  • 对于所有相关的指令,ttxxyy 不必是两两之间不同的。
  • 对于指令 leftrightpp 必须至少为 00 且至多为 bb
  • 对于指令 storevv 的长度必须为 bb

你还可以调用以下函数,以帮助测试你的答案:

void append_print(int t)
  • 在评测你的答案时,对该函数的所有调用都将被忽略。
  • 在评测程序示例中,该函数将往程序追加一个 print(t)\operatorname{\mathit{print}}(t) 操作。
  • 当评测程序示例在执行某个程序过程中遇到一个 print(t)\operatorname{\mathit{print}}(t) 操作时,它会打印出由寄存器 tt 中前 nkn \cdot k 比特构成的 nnkk-比特整数(细节可参见“评测程序示例”部分)。
  • tt 必须满足 0tm10 \le t \le m - 1
  • 对该函数的任何调用,都不会算到你所创建的指令的数量里面。

在追加最后一条指令之后,construct_instructions 应当返回。随后你创建的程序将在一定数量的测试用例上评测,其中每个测试用例给出的输入数据为 nnkk-比特整数 a[0],a[1],,a[n1]a[0], a[1], \ldots , a[n - 1]。如果程序对给定输入数据的输出结果 c[0],c[1],,c[n1]c[0], c[1], \ldots , c[n - 1] 满足如下条件,你的答案就将被视为通过了对应的样例:

  • 如果 s=0s = 0c[0]c[0] 应当为 a[0],a[1],,a[n1]a[0], a[1], \ldots , a[n - 1] 中的最小值。
  • 如果 s=1s = 1,对所有 ii0in10 \le i \le n - 1)来说,c[i]c[i] 应当是 a[0],a[1],,a[n1]a[0], a[1], \ldots , a[n - 1] 中第 1+i1 + i 小的整数。

在评测你的答案时,可能会给出下面的错误信息之一:

  • Invalid index:在调用某些函数时的参数 ttxxyy 所给出的寄存器下标是不正确的(可能是负数)。
  • Value to store is not b bits long:提供给 append_storevv 的长度不等于 bb
  • Invalid shift value:提供给 append_leftappend_rightpp 的值不在 00bb 之间(包括 00bb)。
  • Too many instructions:你的函数试图追加超过 qq 条指令。

评测程序示例

评测程序示例按以下格式读取输入:

  • 11 行:s  n  k  qs \; n \; k \; q

接下来还有若干行,每行描述一个单独的测试用例。每个测试用例将以如下格式给出:

  • a[0]  a[1]    a[n1]a[0] \; a[1] \; \ldots \; a[n - 1]

这样,就描述出了一个包含 nn 个整数 a[0],a[1],,a[n1]a[0], a[1], \ldots , a[n - 1] 的测试用例。在所有测试用例的描述之后,将跟着仅包含 1-1 的单独一行。

评测程序示例首先调用 construct_instructions(s, n, k, q)。如果该调用违反了在程序说明中的某些限制,评测程序示例将打印出在“实现细节”部分列出的错误信息之一,并且退出。否则,评测程序示例首先依次打印出 construct_instructions(s, n, k, q) 所追加的指令。对于 store\operatorname{\mathit{store}} 指令,将把 vv 从下标 00b1b - 1 依次打印出来。

随后,评测程序示例将依次处理测试用例。对于每个测试用例,评测程序示例将在该测试用例中的输入数据上运行所创建的程序。

对于每个 print(t)\operatorname{\mathit{print}}(t) 操作,设 d[0],d[1],,d[n1]d[0], d[1], \ldots , d[n - 1] 为一个整数序列,使得对于所有 ii0in10 \le i \le n - 1),d[i]d[i] 为(在执行该操作时)寄存器 tt 中比特序列 iki \cdot k(i+1)k1(i + 1) \cdot k - 1 的整数值。评测程序按照如下格式打印出该序列:register t: d[0]  d[1]    d[n1]d[0] \; d[1] \; \ldots \; d[n - 1]

如果所有指令都被执行完毕,评测程序示例将打印出程序的输出结果。

如果 s=0s = 0,评测程序示例在每个测试用例上的输出结果为如下格式:

  • c[0]c[0]

如果 s=1s = 1,评测程序示例在每个测试用例上的输出结果为如下格式:

  • c[0]  c[1]    c[n1]c[0] \; c[1] \; \ldots \; c[n - 1]

在处理完所有测试用例后,评测程序打印出 number of instructions: XX,这里 XX 是你的程序的指令数量。

0 2 1 1000
0 0
0 1
1 0
1 1
-1

move 1 0
right 1 1 1
and 0 0 1
0
0
0
1

1 2 1 1000
0 0
0 1
1 0
1 1
-1

move 1 0
right 1 1 1
and 2 0 1
or 3 0 1
left 3 3 1
or 0 2 3
0 0
0 1
0 1
1 1

数据范围与提示

对于所有数据:

  • m=100m = 100
  • b=2000b = 2000
  • 0s10 \le s \le 1
  • 2n1002 \le n \le 100
  • 1k101 \le k \le 10
  • q4000q \le 4000
  • 0a[i]2k10 \le a[i] \le 2^k - 1(对于所有 0in10 \le i \le n - 1
子任务 分值 特殊限制
11 1010 s=0s = 0n=2n = 2k2k \le 2q=1000q = 1000
22 1111 s=0s = 0n=2n = 2k2k \le 2q=20q = 20
33 1212 s=0s = 0q=4000q = 4000
44 2525 s=0s = 0q=150q = 150
55 1313 s=1s = 1n10n \le 10q=4000q = 4000
66 2929 s=1s = 1q=4000q = 4000