AT_sumitb2019_e Colorful Hats 2

题目描述

有 $N$ 个人排成一列,按照从前到后的顺序编号为 $1, 2, 3, \ldots, N$。每个人都戴着一顶帽子,帽子的颜色可以是红色、蓝色或绿色中的一种。 现在,第 $i$ 号的人说: - “在我前面,戴着和我同样颜色帽子的人恰好有 $A_i$ 个。” 假设所有人的发言都是真实的,请你计算 $N$ 个人帽子颜色的所有可能组合数。 由于答案可能非常大,请输出对 $1000000007$ 取模后的结果。

输入格式

输入通过标准输入给出,格式如下: > $N$ $A_1$ $A_2$ $A_3$ $\ldots$ $A_N$

输出格式

输出 $N$ 个人帽子颜色的所有可能组合数,对 $1000000007$ 取模后的结果。

说明/提示

### 限制条件 - $1 \leq N \leq 100000$ - $0 \leq A_i \leq N-1$ - 输入中的所有值均为整数 ### 样例解释 1 可以考虑以下 $3$ 种组合: - 红,红,红,红,红,红 - 蓝,蓝,蓝,蓝,蓝,蓝 - 绿,绿,绿,绿,绿,绿 由 ChatGPT 4.1 翻译