Submission #43122178

---

Source Code Expand

#include<bits/stdc++.h>
#include<atcoder/all>

using namespace std;
using namespace atcoder;
typedef long long ll;

#define rep(i,n) for(ll i=0;i<n;++i)
#define rrep(i,n) for(ll i=n-1;i>=0;--i)
#define FOR(i,s,e) for(ll i=s;i<=e;++i)
#define FFOR(i,s,e) for(ll i=s;i>=e;--i)

#define yesno(flg) if(flg){cout<<"Yes"<<endl;}else{cout<<"No"<<endl;}
#define ALL(a) (a).begin(),(a).end()
#define mp make_pair
#define pb push_back
#define vl vector<ll>
#define vs vector<string>
#define so(a) sort(a.begin(),a.end())

#define fi first
#define se second
#define print(a) cout<<a<<endl
#define ssize(a) (ll)(a.size())

#define MAX_N 1002
#define i197 1000000007
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; }


const ll INF=1000000000000000001;
struct edge
{
	int to, cost;
};

struct pos{
	// 1 変数を入れる;
	ll t;
	ll x,y;
	set<ll> st;

};
int dx[4] = { 1, 0, -1, 0};
int dy[4] = { 0, 1, 0, -1};
int dd[5] = { 0, 1, 0, -1, 0};

typedef pair<ll, ll> P1;
typedef pair<pair<ll,ll>,ll> P3;
typedef pair<string,ll> Ps;
typedef pair<pair<ll,ll>,pair<ll,ll> > P4;

typedef pair<long double,ll> P2;



int main(){

	ios::sync_with_stdio(0);
	cin.tie(0);

	ll n;
	cin>>n;

	vl a(n+5);
	char c[n+5];

	rep(i,n)cin>>a[i+1];
	rep(i,n)cin>>c[i+1];

	vl d1(4);
	ll d[3][3]={};

	ll ans=0;
	FOR(i,1,n){
		if(c[i]=='M'){
			d1[a[i]]++;
		}else if(c[i]=='E'){
			rep(j,3){
				ll j1=j;
				ll j2=a[i];
				if(j2<j1)swap(j1,j2);
				d[j1][j2]+=d1[j];
			}
		}else{
			rep(j,3){
				rep(k,3){
					if(d[j][k]==0)continue;
					vl b(4);
					b[a[i]]=1;
					b[j]=1;
					b[k]=1;
					rep(l,4){
						if(b[l]==0){
							ans+=l*d[j][k];
							break;
						}
					}
				}
			}

		}
	}

	print(ans);










	return 0;
}

Submission Info

Judge Result