【AcWing】4261. 孤独的照片

题目描述

解题思路

求要去除的方案数：长度至少为3的序列中有一个字母只能出现一次。
依此枚举每一个位置的该字母只出现一次的情况下，序列长度不小于3的所有方案数。
我们假设只出现一次的字母为G，G的左右两侧为H。 左侧H的数量为L，右侧为R。
可以分为以下三种情况：

1.G的左右两侧至少都包含一个H     ---> 方案数： L * R
2.G的左侧没有H(右侧至少有两个H)  ---> 方案数： R - 1
3.G的右侧没有H(左侧至少有两个H)  ---> 方案数： L - 1
总方案数：L * R + R - 1 + L - 1

代码实现

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 500010;
int n;
char str[N];
int l[N], r[N]; // l[i] 表示i的位置左边有多少个H， r[i] 表示i的位置右边有多少个H

int main()
{
    scanf("%d", &n); scanf("%s", str);
    // 预处理l
    for (int i = 0, h = 0, g = 0; i < n; i ++)
        if (str[i] == 'G') l[i] = h, h = 0, g ++;
        else l[i] = g, g = 0, h ++;
    // 预处理r    
    for (int i = n - 1, h = 0, g = 0; i >= 0; i --)
        if (str[i] == 'G') r[i] = h, h = 0, g ++;
        else r[i] = g, g = 0, h ++;
        
    LL res = 0;
    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;
}