PAT （甲级）1093

题目

题目链接

思路

把每一个字符以左的P字符个数统计出来，每一个字符以右的T字符个数统计出来，在遍历每一个字符串，如果这个字符为A，那么包含这个A的pat个数为 左边P的个数 * 右边T的个数；
注意点：记得对每次结果取余；

代码

#include <iostream>
#include <string>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
const int maxn = 1e5 + 10;
const int mod = 1000000007;
char s[maxn];
int p[maxn], t[maxn];

int main()
{
     cin.getline(s, maxn);
     int size = strlen(s);
     //初始化首位和末位
     if(s[0] == 'P') p[0] = 1;
     else p[0] = 0;
     if(s[size - 1] == 'T') t[size - 1] = 1;
     else t[size - 1] = 0;
     //计算当前位以左（包括当前位）p的个数
     for(int i = 1; i < size; i ++){
          if(s[i] == 'P') p[i] = p[i - 1] + 1;
          else p[i] = p[i - 1];
     }
     //计算当前位以右（包括当前位）t的个数
     for(int i = size - 2; i >= 0; i --){
          if(s[i] == 'T') t[i] = t[i + 1] + 1;
          else t[i] = t[i + 1];
     }
     int ans = 0;
     for(int i = 0; i < size; i ++){
          if(s[i] == 'A') ans = (ans + p[i] * t[i]) % mod;
     }
     printf("%d\n", ans);
     system("pause");
     return 0;
}