做法:贪心
策略:把蓝色的数和红色的数分开来看,红色的数从大到小看,看是否能从大到小填满n ~ (n - 红色数的数量) ,也就是说能从大到小排时,第i个数能不能填上第 (n - i + 1)个数。
蓝色的数从小到大看,看是否能从小到大填上1 ~ 蓝色数的个数,也就是看第i个数能不能填上i。
总结一下就是:
分开来维护
看蓝色的数是否大于等于它应该在的位置
看红色的数是否小于等于它应该所在的位置
#include<bits/stdc++.h>
using namespace std;
const int N = 2e5 + 10;
int num[N];
bool cmp(int a, int b)
{
return a > b;
}
int main()
{
int t; cin >> t;
while (t -- )
{
vector<int> a, b;
int n; string s;
cin >> n;
for (int i = 1; i <= n; i ++ ) scanf("%d", &num[i]);
cin >> s;
for (int i = 1; i <= n; i ++ )
{
if (s[i - 1] == 'R') a.push_back(num[i]);
if (s[i - 1] == 'B') b.push_back(num[i]);
}
sort (a.begin(), a.end(), cmp);
sort (b.begin(), b.end());
bool f = false;
for (int i = 0; i < b.size(); i ++ )
{
if (b[i] < i + 1)
{
f = 1;
break;
}
}
if (f)
{
puts("NO");
continue;
}
for (int i = 0; i < a.size(); i ++ )
{
if (a[i] > n - i)
{
f = 1;
break;
}
}
if (!f) puts("YES");
else puts("NO");
}
return 0;
}