首先想到直接求最小值,去讨论间隔数和操作数之间的关系,如果间隔数大于操作数,最后一定会有惩罚值出现,但是如何去安排每次操作呢?多次尝试没有找到结果。
然而,如果第一步去考虑能否将惩罚值降到某一个阈值之下呢?如果能解决这一个问题,就可以使用二分解决。
#include <bits/stdc++.h>
using namespace std;
const int N = 3e5 + 10;
const int mod = 1e9 + 7;
int n, k;
string s;
int a[N];
bool check(int x)
{
int ans = 0, cnt = 0;
for(int i = 0; i < n; i ++ )
{
if(a[i] <= x) continue;
if(s[i] == 'R' && cnt) cnt = 0, ans ++;
if(s[i] == 'B') cnt ++;
}
if(cnt) ans ++;
if(ans <= k) return true;
else return false;
}
void solve()
{
cin >> n >> k >> s;
for(int i = 0; i < n; i ++ ) cin >> a[i];
int l = 0, r = *max_element(a, a + n);
while(l < r)
{
int mid = l + r >> 1;
if(check(mid)) r = mid;
else l = mid + 1;
}
cout << l << "\n";
return;
}
int main()
{
int t;
cin >> t;
while(t -- ) solve();
return 0;
}