div4的G题,对我来说还是有难度的。方法很好,思维题。但我看了题解才想到。
思路
字符串s
只有这4中情况:
- 若干个B + 若干个A + … + 若干个A + 若干个B
例如,BBBAABAAAB - 若干个B + 若干个A + … + 若干个A
例如,BBBAABAAA - 若干个A + … + 若干个A + 若干个B
例如,AABAAAB - 若干个A + … + 若干个A
例如,AAABABABABBAAAA
对于前3中情况,自然所有的A都能转化,即 res = 字符A的数量。
对于第四种情况,如果B都是不连续的,必然有且仅有一段连续的A得不到转换。此时 res = 字符A的数量 - 最短的连续的A的数量
#include <bits/stdc++.h>
#define int long long
#define sz(a) ((int)a.size())
#define all(a) a.begin(), a.end()
using namespace std;
using PII = pair<int, int>;
using i128 = __int128;
const int N = 2e5 + 10;
int n;
void solve() {
string s;
cin >> s;
n = sz(s);
s.insert(s.begin(), '.');
int num_a_parts = 0;
int min_a_part = 1e18;
int num_a = 0;
int num_b = 0;
for (int i = 1; i <= n; i ++) {
if (s[i] == 'B') {
num_b ++;
} else {
int r = i;
while (r < n && s[r + 1] == 'A') {
r ++;
}
num_a_parts ++;
min_a_part = min(min_a_part, r - i + 1);
num_a += r - i + 1;
i = r;
}
}
if (num_b >= num_a_parts) {
cout << num_a << "\n";
} else {
cout << num_a - min_a_part << "\n";
}
}
signed main() {
ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
int T = 1;
cin >> T; cin.get();
while (T --) solve();
return 0;
}
```(https://codeforces.com/contest/1873/problem/G)
[参考文章](https://zhuanlan.zhihu.com/p/657837614)
div4的G题,对我来说还是有难度的。方法很好,思维题。但我看了题解才想到。
# 思路
字符串`s`只有这4中情况:
1. 若干个B + 若干个A + ... + 若干个A + 若干个B
例如,BBBAABAAAB
2. 若干个B + 若干个A + ... + 若干个A
例如,BBBAABAAA
3. 若干个A + ... + 若干个A + 若干个B
例如,AABAAAB
4. 若干个A + ... + 若干个A
例如,AAABABABABBAAAA
对于前3中情况,自然所有的A都能转化,即 res = 字符A的数量。
对于第四种情况,必然有且仅有一段连续的A得不到转换。此时 res = 字符A的数量 - 最短的连续的A的数量
```cpp
#include <bits/stdc++.h>
#define int long long
#define sz(a) ((int)a.size())
#define all(a) a.begin(), a.end()
using namespace std;
using PII = pair<int, int>;
using i128 = __int128;
const int N = 2e5 + 10;
int n;
void solve() {
string s;
cin >> s;
n = sz(s);
s.insert(s.begin(), '.');
int num_a_parts = 0;
int min_a_part = 1e18;
int num_a = 0;
int num_b = 0;
for (int i = 1; i <= n; i ++) {
if (s[i] == 'B') {
num_b ++;
} else {
int r = i;
while (r < n && s[r + 1] == 'A') {
r ++;
}
num_a_parts ++;
min_a_part = min(min_a_part, r - i + 1);
num_a += r - i + 1;
i = r;
}
}
if (num_b >= num_a_parts) {
cout << num_a << "\n";
} else {
cout << num_a - min_a_part << "\n";
}
}
signed main() {
ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
int T = 1;
cin >> T; cin.get();
while (T --) solve();
return 0;
}