1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
|
#include<bits/stdc++.h> using namespace std; using ll=long long; #define rep(i,a,n) for(int i=a;i<=n;i++) #define frep(i,a,n) for(int i=a;i>=n;i--) #define int long long const int N=1e5+100; #define PII pair<int,int> map<PII,int>mp; int a[N]; int b[N]; signed main() { int n; cin>>n; rep(i,1,n) { cin>>a[i]; } rep(i,1,n) { char s; cin>>s; if(s=='B') { mp[{1,a[i]}]++; } else { mp[{0,a[i]}]++; } } int ans=0; for(auto& [P,cnt]:mp) { auto [x,y]=P; ans+=mp[{1-x,y}]*mp[{x,y}]; } cout<<ans/2; return 0; }
|