#L4277. 「KTSC 2022 R2」彩色括号序列

「KTSC 2022 R2」彩色括号序列


注意事项

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

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

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


题目描述

题目译自 2022년도 국제정보올림피아드 대표학생 선발고사 - 2차 선발고사 T2 「알록달록한 괄호열」

括号序列是由两种字符 () 组成的字符串。好的括号序列可以通过以下规则构成:

  1. 空字符串是好的括号序列。
  2. 如果 SS 是好的括号序列,那么 (S)(S) 也是好的括号序列。在这种情况下,SS 两端的括号是配对的。
  3. 如果 SSTT 都是好的括号序列,那么 STST 也是好的括号序列。

彩色括号序列是指每个括号都被涂上特定颜色的括号序列。五彩缤纷的括号序列需要满足以下所有条件:

  • 忽略颜色,仅看括号的形式时是好的括号序列。
  • 所有相邻的两个括号颜色不同。
  • 所有配对的两个括号颜色不同。

当从字符串 SS 中选出一个或多个字符按顺序排列成 TT 时,称可以从 SS 中选出 TT。给定一个彩色括号序列,问可以从中选出多少个五彩缤纷的括号序列?即使括号形式相同,但只要颜色不同就算作不同情况;即使选取方式不同,但结果相同也只算作一种情况。


实现细节

你需要实现以下函数:

int count(vector<int> P);
  • 该函数只会被调用一次。
  • PP:表示彩色括号序列,P[i]P[i] 表示第 ii 个括号的信息。P[i]<0P[i] < 0 表示左括号,P[i]>0P[i] > 0 表示右括号,P[i]|P[i]| 的值表示颜色。
  • 该函数需要返回从给定彩色括号序列中可以选出的五彩缤纷括号序列的数量,并对 10000000071000000007 取模。

注意,提交的代码中不应包含任何输入输出操作。


样例 1

用颜色 cc 涂的括号 pp 表示为 pcp_c。给定彩色括号序列
()()()123312\underset{123312}{()()()},评测程序会调用如下函数:

count({-1, 2, -3, 3, -1, 2})

可以选出的五彩缤纷括号序列有

  • ()12()_12()13()_13()32()_32
  • ()()1212()()_{1212}()()1232()()_{1232}()()1312()()_{1312}

66 种。因此,调用的 count 函数应返回 66

从中选取特定字符串的方法有多种,在这个样例中,选取 ()12()_12 的方法有 33 种。


样例 2

给定彩色括号序列 (()())(()()),评测程序会调用如下函数:

count({1, 6, -3, -6, 1, -1, 3, 6})

可以选出的五彩缤纷括号序列有

  • ()13()_13()16()_16()31()_31()36()_36()61()_61()63()_63
  • (())1313(())_{1313}(())1316(())_{1316}(())1613(())_{1613}(())1616(())_{1616}(())1636(())_{1636}(())3136(())_{3136}(())3616(())_{3616}(())3636(())_{3636}
  • ()()1613()()_{1613}()()1616()()_{1616}()()1631()()_{1631}()()1636()()_{1636}
  • ()(())163136()(())_{163136}()(())163616()(())_{163616}()(())163636()(())_{163636}

2121 种。因此,调用的 count 函数应返回 2121

(())1336(())_{1336}(())6136(())_{6136} 虽然括号序列是好的,但由于相邻两个括号颜色相同或配对括号颜色相同,因此不是五彩缤纷的括号序列。


样例 3

给定彩色括号序列
(()())())()()))714126347562657\underset{714126347562657}{(()())())()()))},评测程序会调用如下函数:

count({-7, -1, 4, -1, 2, 6, -3, 4, 7, -5, 6, -2, 6, 5, 7})

调用的 count 函数应返回 11161116


数据范围与提示

对于所有输入数据,满足:

  • NN 表示 PP 的长度,1N7001 \leq N \leq 700
  • 对于所有 ii (0iN1)(0 \leq i \leq N-1)1P[i]N1 \leq |P[i]| \leq N

详细子任务附加限制及分值如下表所示。

子任务 分值 附加限制
1 5 N20N \leq 20
2 30 N200N \leq 200
3 27 N500N \leq 500;对于所有 ii (0iN1)(0 \leq i \leq N-1)P[i]20\lvert P[i]\rvert \leq 20
4 14 对于所有 ii (0iN1)(0 \leq i \leq N-1)P[i]2\lvert P[i]\rvert \leq 2
5 24 无附加限制

示例评测程序

示例评测程序的输入格式如下:

第 1 行:N
第 2 行:P[0] P[1] ... P[N-1]

示例评测程序的输出格式如下:

第 1 行:count 函数返回的值