GOJ 1136 查找ACM(加强版) (双指针的应用)

查找ACM(加强版)

Time Limit: 2000/1000ms (Java/Others)

Problem Description:
作为一个acmer，应该具备团队合作能力和分析问题能力。给你一个只有a，c和m的字符串，你要依次取3个字母使之恰好为acm。
比如串
accmmmca 你可以取
12345678
ac_m____
ac__m___
ac___m__
a_cm____
a_c_m___
a_c__m__共6种。

你只要给出给你的串有多少种方案能组成acm。
Input:

输入包含多组测试，对于每组测试，输入一个只有acm3种字母的串（长度<=200000）

Output:

对于每组测试，输出一个整数一行，表示给你的串有多少种方案能组成acm。
Sample Input:

accmmmca
Sample Output:
6

PS: 因为要查找的字符串的数量级为 1e5, 所有不能每找一个 ‘c’ , 就遍历整个串，这样的时间复杂度为O(n^2)，会TLE 的。 为了降低时间复杂度，可以用一个数组保存当前字符（包括它本身）的前面有多少个’a’。然后从后面开始遍历整个串，如果遇到’c’, 就累加此处的组成方案。这样可以将时间复杂度降低为O(n).

#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 200004
using namespace std;
int a[maxn];
int main()
{
    ios::sync_with_stdio(false);
    char s[maxn];
    while(cin>>s)
    {
        int n = strlen(s);
        fill(a, a + n, 0);
        for(int i = 0; i < n; i++)
        {
            if(s[i] == 'a' && i != 0) a[i] = a[i - 1] + 1;
            else if(s[i] != 'a' && i != 0) a[i] = a[i - 1];
            else if(i == 0 && s[i] == 'a')a[i] = 1;
            else a[i] = 0;
        }
        long long m_num = 0;
        long long ans = 0; //注意要使用long long 
        for(int i = n - 1; i >= 0; i--)
        {
            if(s[i] == 'm')m_num++;
            if(s[i] == 'c')
            {
                ans += a[i] * m_num;
            }
        }
        cout<<ans<<endl;
    }
    return 0;
}