奶牛贝茜在她最喜欢的牧场中发现了一块石碑,上面刻有神秘的碑文。
碑文的文字似乎来自一种神秘的古代语言,可看作一个只包含 C,O,W 三种字符的字符串。
尽管贝茜无法解密该文字,但是她很欣赏 C,O,W 按顺序构成她最喜欢的单词 COW。
她想知道 COW 在碑文中一共出现了多少次。
她不介意 C,O,W 之间是否存在其他字符,只要这三个字符按正确的顺序出现即可。
她也不介意多个不同的 COW 是否共享了一些字符。
例如,COW 在 CWOW 中只出现一次,在 CCOW 中出现两次,在 CCOOWW 中出现八次。
给定碑文中的文字,请帮助贝茜计算 COW 出现的次数。
输入格式
第一行包含 N。
第二行包含一个长度为 N 的字符串,其中只包含字符 C,O,W。
输出格式
输出给定字符串中 COW 作为子序列(不一定连续)的出现次数。
数据范围
1≤N≤105
输入样例:
6
COOWWW
输出样例:
6
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100010;
typedef long long LL;
char str[N];
int main(){
int n;
cin >> n;
for (int i = 0; i < n; i ++ ){
cin >> str[i];
}
LL a = 0,b = 0,c = 0;
for (int i = 0; i < n; i ++ ){
if(str[i] == 'C') a++;
else if(str[i] == 'O') b += a;
else if(str[i] == 'W')c += b;
}
cout << c;
return 0;
}
关键:状态机(算法题高课)
关键在于这个for循环的理解与应用
for (int i = 0; i < n; i ++ ){
if(str[i] == 'C') a++;
else if(str[i] == 'O') b += a;
else if(str[i] == 'W')c += b;
}