当前位置: 代码迷 >> 综合 >> CF1607D Blue-Red Permutation
  详细解决方案

CF1607D Blue-Red Permutation

热度:41   发布时间:2023-10-14 01:09:18.0

做法:贪心
策略:把蓝色的数和红色的数分开来看,红色的数从大到小看,看是否能从大到小填满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;
}
  相关解决方案