4261. 孤独的照片

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\times R$
  • L=0且R>2时，情况总数为 $R-1$
  • L>2且R=0时，情况总数为 $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;
}