AT_abc308_e [ABC308E] MEX
题目描述
给定一个由 $0,1,2$ 组成的长度为 $N$ 的数列 $A=(A_1,A_2,\dots,A_N)$,以及一个由 `M`、`E`、`X` 组成的长度为 $N$ 的字符串 $S=S_1S_2\dots S_N$。
请计算所有满足 $1 \leq i < j < k \leq N$ 且 $S_iS_jS_k=$ `MEX` 的整数三元组 $(i,j,k)$,对于每个三元组,求 $\text{mex}(A_i,A_j,A_k)$ 的总和。这里,$\text{mex}(A_i,A_j,A_k)$ 表示不等于 $A_i,A_j,A_k$ 的最小非负整数。
输入格式
输入通过标准输入给出,格式如下:
> $N$ $A_1$ $A_2$ $\dots$ $A_N$ $S$
输出格式
请输出答案,结果为一个整数。
说明/提示
## 限制条件
- $3 \leq N \leq 2 \times 10^5$
- $N$ 为整数
- $A_i \in \{0,1,2\}$
- $S$ 是由 `M`、`E`、`X` 组成的长度为 $N$ 的字符串
## 样例解释 1
满足 $S_iS_jS_k = $ `MEX` 的 $(i,j,k)$ 共有 $(1,2,4),(1,3,4)$ 两组。$\text{mex}(A_1,A_2,A_4)=\text{mex}(1,1,2)=0$,$\text{mex}(A_1,A_3,A_4)=\text{mex}(1,0,2)=3$,因此答案为 $0+3=3$。
由 ChatGPT 4.1 翻译