Description
小伟暑假期间到大草原旅游,在一块石头上发现了一些有趣的碑文。碑文似乎是一个神秘古老的语言,只包括三个大写字母 C、O 和 W。尽管小伟看不懂,但是令他高兴的是,C、O、W的顺序形式构成了一句他最喜欢的奶牛单词“COW”。现在,他想知道有多少次 COW 出现在文本中。 如果 COW 内穿插了其他字符,只要 COW 字符出现在正确的顺序,小伟也不介意。甚至,他也不介意出现不同的 COW 共享一些字母。例如,CWOW 出现了 1 次 COW,CCOW 算出现了2 次 COW,CCOOWW 算出现了 8 次 COW。
Input
第 1 行为 1 个整数 N。 第 2 行为 N 个字符的字符串,每个字符是一个 C、O 或 W。
Output
输出 COW 作为输入字符串的字串出现的次数(不一定是连续的)。 提示:答案会很大,建议用 64 位整数(long long)。
Accepted Code
#include <bits/stdc++.h> using namespace std; char ch[100005]; typedef long long LL; int main() { int t = 0; cin >> t; memset(ch, 0, sizeof(ch)); scanf("%s", ch); LL cur_c = 0; LL cur_co = 0; LL cur_cow = 0; for (int i = 0; i < t; i++) { if (ch[i] == 'C') { cur_c++; } else if (ch[i] == 'O') { cur_co += cur_c; } else if (ch[i] == 'W') { cur_cow += cur_co; } } cout << cur_cow << endl; }
解后反思
本题就比较有趣,利用向后遍历的过程中的状态转移加以优化。
例如在一串字符串中COWCW
第一次读到C cur_c=1 cur_co=0 cur_cow=0
第二次读到O cur_c=1 cur_co+=cur_c=1 cur_cow=0
第二次读到W cur_c=1 cur_co=1 cur_cow+=cur_co=1
以此类推
最终可得所有的COW有2个