ABC308E（前后缀和）

题目传送门： 

​​​​​​https://atcoder.jp/contests/abc308/tasks/abc308_e

 

E-MEX

题意：

一个大小为n的由{0，1，2}组成的数组a和长度为n的由{M，E，X}组成的字符串，求每组满足i<j<k且Si=M，Sj=E，Sk=X的mex(a[i],a[j],a[k])的和。

题解：

本质是前后缀和，由MEX的构成可知，E处中间，由此联想到将每个E前面的所有M和E后面的所有X在0，1，2三种情况下的个数分别求出来，然后对应求mex，最后求和即可。因此，可用前缀和求出在要求E前面的M（0，1，2三种情况）的个数，用后缀和求出要求E后面的X（0，1，2三种情况）的个数，然后排列组合出所有情况求mex。

每个E的mex总和为：

可手动打表一下样例：

代码显示如下：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 2e5 + 10;
int a[N], b[N][3], c[N][3];
int n, m, k, g, x, flag;
char s[N];
int mex(int a,int b,int c)//求mex
{
	int t[3]={0};
	++t[a],++t[b],++t[c];
	if(!t[0]) return 0;
	if(!t[1]) return 1;
	if(!t[2]) return 2;
	return 3;
}
void solve() {
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i];
	cin>>(s+1);
	for(int i=1;i<=n;i++)//前缀和求M在对应i的三种情况的个数
	{
		for(int j=0;j<3;j++)
			b[i][j]=b[i-1][j];
		if(s[i]=='M') ++b[i][a[i]];
	}
	
	for(int i=n;i>=1;i--)后缀和求X在对应i的三种情况的个数
	{
		for(int j=0;j<3;j++)
			c[i][j]=c[i+1][j];
		if(s[i]=='X') ++c[i][a[i]];
	}
	int ans=0;
	for(int i=2;i<n;i++)//求所有E的mex总和的和
	{
		if(s[i]=='E')//求每个E的mex总和（对应公式）
		{
			for(int j=0;j<3;j++)
			{
				for(int k=0;k<3;k++)
				{
					ans+=b[i][j]*c[i][k]*mex(j,k,a[i]);
				}
			}
		}
	}
	cout<<ans<<endl;
}
signed main() {
	int t = 1;
//	cin >> t;
	while (t--) {
		solve();
	}
}