【无标题】

游游的数组染色

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网

时间限制：C/C++ 1秒
空间限制：C/C++ 262144K

题目描述

游游拿到了一个数组，其中一些数被染成红色，一些数被染成蓝色。
游游想知道，取两个不同颜色的数，且它们的数值相等，有多少种不同的取法？

我们定义，两种取法如果取的某个数在原数组的位置不同，则定义为不同的取法。

输入描述:

第一行输入一个正整数 n，代表数组的长度。
第二行输入 n 个正整数 ai​，代表数组中的元素。
第三行输入一个长度为 n ，仅包含 'R'和 'B' 两种字符的字符串，第 i 个字符为 ′R′代表数组第 i 个元素被染成红色，为 'B' 代表被染成蓝色。
1≤n≤200000
1≤ai​≤10^9

输出描述:

输出一个整数，代表取数的方案数量。

示例1

输入

5
1 2 1 2 2
BRRBB

Copy

输出

3

Copy

说明

第一种取法，取第一个数和第三个数，颜色不同且都是1。
第二种取法，取第二个数和第四个数，颜色不同且都是2。
第三种取法，取第二个数和第五个数，颜色不同且都是2。

示例2

输入

3
2 3 3
BBB

Copy

输出

0

Copy

说明

所有数都是蓝色，显然取不出两个颜色不同的数。

#include<iostream>
#include<vector>
#include<unordered_map>
using namespace std;

const int N = 2e5+10;
int n, p[N];
char s[N];
unordered_map<int, pair<int, int>>_map;

int main(){
    
    cin >> n;
    for(int i = 0; i < n; ++i){
        cin >> p[i];
    }
    for(int i = 0; i < n; ++i){
        cin >> s[i];
    }
    for(int i = 0; i < n; ++i){
        if(s[i] == 'B'){
            _map[p[i]].first++;
        }
        else _map[p[i]].second++;
    }
    int ans=0;
    for (auto t:_map) {
        ans+=t.second.first*t.second.second;
    }
    cout<<ans<<endl;

    
    return 0;
}