哈尔滨理工大学软件与微电子学院程序设计竞赛(新生赛) S 小乐乐与字符串

题目链接：https://ac.nowcoder.com/acm/contest/1877/S

小乐乐与字符串

题目描述
在庆祝祖国母亲70华诞之际，老师给小乐乐出了一个问题。大家都知道China的英文缩写是CHN，那么给你一个字符串s，你需要做的是统计s中子串“CHN”的个数。
子串的定义：存在任意下标a < b < c，那么“s[a]s[b]s[c]”就构成s的一个子串。如“ABC”的子串有“A”、“B”、“C”、“AB”、“AC”、“BC”、“ABC”。

输入描述:
输入只包含大写字母的字符串s。(1 ≤ length ≤ 8000)
输出描述:
输出一个整数，为字符串s中字串“CHN”的数量。
示例1

输入
CCHNCHN

输出
7
示例2

输入
CCHNCHNCHNCHN

输出
30

思路：这道题一看就是可以用暴力做，但是因为字符串的长度为8000，如果用三重循环的话会超时，所以需要优化。我们可以枚举’C’和’H’，接着其实我们就只需要求’H’以后的区间中出现了多少个’N’，求区间内出现次数，就想到了可以用前缀和来优化。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
 
#define maxn 100010
#define ll long long
#define inf 0x3f3f3f3f
#define mod 1000000007
 
char s[10100];
int a[maxn];
 
int main()
{
    cin>>s;
    int l=strlen(s);
    int sum=0;
    for(int i=2; i<l; i++)
    {
        if(s[i]=='N')
            a[i]=a[i-1]+1;
        else a[i]=a[i-1];
    }
    ll ans=0;
    for(int i=0; i<l; i++)
    {
        if(s[i]=='C')
        {
            for(int j=i+1; j<l; j++)
            {
                if(s[j]=='H')
                {
                    ans+=(a[l-1]-a[j]);
                }
            }
        }
    }
    cout<<ans<<endl;
    return 0;
}
}