103084: [Atcoder]ABC308 E - MEX

Memory Limit:256 MB Time Limit:2 S
Judge Style:Text Compare Creator:
Submit:3 Solved:0

Description

Score : $475$ points

Problem Statement

You are given a length-$N$ sequence $A=(A_1,A_2,\dots,A_N)$ consisting of $0$, $1$, and $2$, and a length-$N$ string $S=S_1S_2\dots S_N$ consisting of M, E, and X.

Find the sum of $\text{mex}(A_i,A_j,A_k)$ over all tuples of integers $(i,j,k)$ such that $1 \leq i < j < k \leq N$ and $S_iS_jS_k=$ MEX. Here, $\text{mex}(A_i,A_j,A_k)$ denotes the minimum non-negative integer that equals neither $A_i,A_j$, nor $A_k$.

Constraints

  • $3\leq N \leq 2\times 10^5$
  • $N$ is an integer.
  • $A_i \in \lbrace 0,1,2\rbrace$
  • $S$ is a string of length $N$ consisting of M, E, and X.

Input

The input is given from Standard Input in the following format:

$N$
$A_1$ $A_2$ $\dots$ $A_N$
$S$

Output

Print the answer as an integer.


Sample Input 1

4
1 1 0 2
MEEX

Sample Output 1

3

The tuples $(i,j,k)\ (1 \leq i < j < k \leq N)$ such that $S_iS_jS_k$ = MEX are the following two: $(i,j,k)=(1,2,4),(1,3,4)$. Since $\text{mex}(A_1,A_2,A_4)=\text{mex}(1,1,2)=0$ and $\text{mex}(A_1,A_3,A_4)=\text{mex}(1,0,2)=3$, the answer is $0+3=3$.


Sample Input 2

3
0 0 0
XXX

Sample Output 2

0

Sample Input 3

15
1 1 2 0 0 2 0 2 0 0 0 0 0 2 2
EXMMXXXEMEXEXMM

Sample Output 3

13

Input

题意翻译

给定长度为 $N$ 的包含 $0$,$1$,$2$ 的序列 $A_1$,$A_2 \dots A_n$,和一个长度为 N 的包含字符 ``MEX`` 的字符串$S = S_1$,$S_2 \dots S_n$。对于所有符合条件 $1 \le i < j < k \le N$,$S_iS_jS_k$ = ``MEX`` 的三元组 $(i, j, k)$,请你求出 $mex(A_i, A_j, A_k)$ 之和。$mex()$ 函数表示未出现在序列中的最小非负整数。

Output

分数:$475$分

问题描述

给你一个由$0$、$1$和$2$组成的长度为$N$的序列$A=(A_1,A_2,\dots,A_N)$,以及一个长度为$N$的字符串$S=S_1S_2\dots S_N$,由MEX组成。

找出所有整数元组$(i,j,k)$的$\text{mex}(A_i,A_j,A_k)$的和,其中$1 \leq i < j < k \leq N$且$S_iS_jS_k=$ MEX。这里,$\text{mex}(A_i,A_j,A_k)$表示等于$A_i$、$A_j$和$A_k$的最小非负整数。

限制条件

  • $3\leq N \leq 2\times 10^5$
  • $N$是一个整数。
  • $A_i \in \lbrace 0,1,2\rbrace$
  • $S$是一个长度为$N$的字符串,由MEX组成。

输入

输入数据从标准输入给出,如下所示:

$N$
$A_1$ $A_2$ $\dots$ $A_N$
$S$

输出

将答案作为整数输出。


样例输入 1

4
1 1 0 2
MEEX

样例输出 1

3

满足$S_iS_jS_k$ = MEX的元组$(i,j,k)\ (1 \leq i < j < k \leq N)$如下两个:$(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$。


样例输入 2

3
0 0 0
XXX

样例输出 2

0

样例输入 3

15
1 1 2 0 0 2 0 2 0 0 0 0 0 2 2
EXMMXXXEMEXEXMM

样例输出 3

13

HINT

mex(a, b, c)表示abc三个数中未出现过的自然数,请问所有满足对应字符是MEX的下标的mex之和是多少?

加入题单

算法标签: