牛客 DP36 abb

题意:

leafee 最近爱上了 abb 型语句,比如“叠词词”、“恶心心”

leafee 拿到了一个只含有小写字母的字符串,她想知道有多少个 "abb" 型的子序列?
定义: abb 型字符串满足以下条件:

  1. 字符串长度为 3 。
  2. 字符串后两位相同。
  3. 字符串前两位不同。

  1. 牛客 DP36 abb_第1张图片

牛客 DP36 abb_第2张图片

牛客 DP36 abb_第3张图片

还以为只有a,b,c三个字符, WA了一发

对于一个字符,可以处理出来它后面的25种字母的个数(从后往前预处理)

#include
#include
using namespace std;
typedef long long LL;
const int N = 1e5 + 10;
int f[30][N];
int n;
string s;
int main()
{
	cin >> n >> s;
	s = " " + s;
	//预处理
	for (int i = n; i >= 1; i--) {
		for (int k = 1; k <= 26; k++) {
			f[k][i] = f[k][i + 1];
		}
		int x = s[i] - 'a' + 1;
		f[x][i]++;
	}
    //处理答案
	LL ans = 0;
	for (int i = 1; i <= n; i++) {
		int x = s[i] - 'a' + 1;
		for (int k = 1; k <= 26; k++) {
			if (x != k) {
				LL res = f[k][i + 1];
				ans += (res - 1) * res / 2;
			}
		}
	}
	cout << ans << endl;
	return 0;
}

你可能感兴趣的:(刷算法题,算法,c++,数据结构)