4261. 孤独的照片
https://www.acwing.com/problem/content/description/4264/
思路
-
首先想到的是爆搜,遍历每个长度大于等于3的串,但数据为500000,时间复杂度过大
-
我没想到的是可以考虑需要删除串的特点:
- 长度大于等于3
- 包含1个G或N(记为S)
- 其余字符全部相同
根据这个特点,我们可以遍历整个字符串,将遍历的字符作为S,考虑S左右两部分的情况(记为L和R
- L>1且R>1时,情况总数为 L × R L \times R L×R
- L=0且R>2时,情况总数为 R − 1 R-1 R−1
- L>2且R=0时,情况总数为 L − 1 L-1 L−1
接下来需要考虑如何快速获得S的L和R长度:
- 可以预处理两个数组L[]和R[]
- 当S为G时,
L[i]
代表左边连续长度为H的长度;当S为H时,L[i]
代表左边连续长度为H的长度 - R[]数组的意义与L[]类似
C++代码
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long ll;
const int N = 500010;
char str[N];
int l[N], r[N];
int n;
ll res;
int main() {
cin >> n >> str;
for (int i = 0, g = 0, h = 0; i < n; i ++) {
if (str[i] == 'G') {
l[i] = h;
g ++;
h = 0;
} else {
l[i] = g;
g = 0;
h ++;
}
}
for (int i = n-1, g = 0, h = 0; i >= 0; i --) {
if (str[i] == 'G') {
r[i] = h;
g ++;
h = 0;
} else {
r[i] = g;
g = 0;
h ++;
}
}
for (int i = 0; i < n; i ++)
res += (ll)l[i] * r[i] + max(l[i] - 1, 0) + max(r[i] - 1, 0);
cout << res << endl;
return 0;
}