B. Chess Cheater
题意
给了你一个字符串,'W’表示赢,‘L’表示输,赢可以加1分,如果赢的前一场也是赢,那么此时可以加两份,现在你可以修改k个地方,问你最高的分数。
思路
贪心
首先我们肯定是保证连胜的场次越多越好,最好是全部都能够赢,但是收到了k的限制,所以我们应该优先让两个’W’之间的’L’变为’W’,这样我们就可以制造尽可能多的连胜。那么假设现在我们知道了胜利的场次,怎样计算分数呢?假设现在胜利的场次为win,假设我们每一场都能够获得两分,那么现在的分数就是2*win,但是有些不是连续的,现在我们要记录不是连续的的场次,我们设没有连胜的场次为cnt_str,那么这里有两种情况:i == 0 || s[i - 1] == 'L'
,这这两种情况下,都属于不是连胜的,那么我们最终的得分就是2×win - cnt_str。
那么接下来我们要找输掉了的场次,对于输掉了的场次也要分情况:首先我们要记录输掉了的场次,还要判断连输的场次,因为我们后面在修改场次的时候都是优先修改不是连输的、连输少的,这样贪的结果才是最优的。
接下来我们对于第一场和最后一场要进行特判,如果这两个位置都是输的话,我们要把它们放到最后,然后进行排序,(见代码);
#include <bits/stdc++.h>
using namespace std;
const int inf = 0x3f3f3f3f;
void solve()
{
int n, k; cin >> n >> k;
string s; cin >> s;
int win = 0, los = 0, win_str = 0;
vector<int> los_str;
for (int i = 0; i < n; i ++)
{
if (s[i] == 'W')
{
if (i == 0 || s[i - 1] == 'L') win_str ++;
win ++;
}
else
{
los ++;
if (i == 0 || s[i - 1] == 'W') los_str.push_back(0);
los_str.back() ++;
}
}
if (k >= los) { cout << 2 * n - 1 << endl; return; }
if (win == 0)
{
if (k == 0) cout << 0 << endl;
else cout << 2 * k - 1 << endl;
}
if (s[0] == 'L') los_str[0] = inf;
if (s[n - 1] == 'L') los_str.back() = inf;
sort(los_str.begin(), los_str.end());
int len = los_str.size();
win += k;
for (int i = 0; i < len; i ++)
{
if (los_str[i] > k) break;
else { win_str --; k -= los_str[i]; }
}
cout << 2 * win - win_str << endl;
}
signed main()
{
std::ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int t; cin >> t;
while (t --) solve();
return 0;
}