csp模拟3rd_C - 咕咕东学英语

C - 咕咕东学英语

---

题目描述

题目思路

其实刚看这道题的时候，一直看不懂题意…也以为这个题很麻烦。可是再我问了大佬之后才发现虽然题意有点难理解，但是做起来真的就不难。通过观察找规律之后我们发现不是delicious的串都有一个特点，就是出现了AA…AB…/BB…BA…这两种情况。所以要想求得delicious串的数目可以用所有的子串减去不是delicious的串的个数。其中在AA…AB…/BB…BA…这两种结构下不是delicious子串的个数就是其中A/B的个数。接下来就很简单了，不过需要注意我们需要从前后两次遍历输入的字符串，因为出现不是delicious的子串时如果只是一遍遍历的话就不能找出所有的不是delicious的子串，比如AAAABB串中一次遍历只能找出AAAAB这种的子串却无法减去ABB这不是delicious的子串。这个题目告诉我们，题意难懂，解题思路其实可能很简单。

代码实现

#include <iostream>
#include <string>
using namespace std;

#define _for(i,a,b) for(int i = (a); i < (b); i++)
#define _rep(i,a,b) for(int i = (a); i <= (b); i++)

long long n,ans;
string s;

int main(){
	cin >> n >> s;
	ans = (n*(n-1))/2;
	int flag = 0, tmp = 0;
	_for(i,0,n-1){
		if(s[i] != s[i+1]){
			ans -= (i+1-flag);
			flag = i+1;
			tmp++;
		}
	}
	flag = n-1;
	ans += tmp;
	for(int i = n-1; i > 0; i--){
		if(s[i] != s[i-1]){
			ans -= (flag-i+1);
			flag = i-1;
		}
	}
	
	cout << ans << endl;
}