CF1607D Blue-Red Permutation

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

猜你喜欢

转载自blog.csdn.net/qq_34682765/article/details/121500356
今日推荐