【数学】字符串

题目描述

小熊有一个由小写英文字母组成的字符串$s=s1,s2...sn$。小熊想要计算s中有多少子串包含字符串“bear”，也就是找出满足字符串$x(i,j)=si,si+1\dots sj$ 包含至少一个字符串“bear”的 ($i,j$)对数（$1\le i\le j\le n$）。
字符串$x(i,j)$包含字符串“bear”定义为存在一个整数k（i≤k≤j-3），满足sk=b，sk+1=e，sk+2=a，sk+3=r。
请帮助小熊解决这个问题。

输入

输入共1行，包含一个非空字符串s。数据保证字符串s中只包含小写英文字母。

输出

输出共1行，包含一个整数，表示这个问题的答案。

输入样例

bebearar

输出样例复制

9

说明

【输入输出样例说明】

符合条件的9对(i, j)为：(1，6)，(1，7)，(1，8)，(2，6)，(2，7)，(2，8)，(3，6)，(3，7)，(3，8)。

【数据说明】

对于$50\%$的数据，$1\le n\le 200$。
对于$100\%$的数据，$1\le n\le 3000$。

---

解题思路

选出区间$[i,j]$，保证去区间里有一个bear

先找出所有的bear，记录它(们)的位置
区间左端点是（1 ~ b的位置），区间右端点是（r的位置 ~ 下一个bear的a位置）
那么区间可能数就是（b的位置）* （下一个bear的a位置 - r的位置 + 1）

为什么右端点不是n呢？
因为会和下一个bear的区间重合，就是一个去重的作用

为了保证最后一个bear能正确计算，要在n+1的位置建一个墙

---

Code

#include <iostream>
#include <cstdio>
#include <string>
using namespace std;
string s;
int b[3200], num, r[3200];
long long ans;
int main(){
	cin>>s;
	for (int i = 0; i <= s.size() - 4; i++)
		if (s[i] == 'b' && s[i + 1] == 'e' && s[i + 2] == 'a' && s[i + 3] == 'r')
			b[++num] = i + 1, r[num] = i + 4, i += 3;//记录bear的位置，要+是因为字符串是从0开始存储的
	r[++num] = s.size() + 1;//建墙
	for (int i = 1; i < num; i++)
	    ans = b[i] * ((r[i + 1] - 1) - r[i] + 1) + ans; //公式计算
	printf ("%lld", ans);
}