1040. 有几个PAT(25)
题目
字符串APPAPT中包含了两个单词“PAT”,其中第一个PAT是第2位(P),第4位(A),第6位(T);第二个PAT是第3位(P),第4位(A),第6位(T)。
现给定字符串,问一共可以形成多少个PAT?
输入格式
输入只有一行,包含一个字符串,长度不超过105,只包含P、A、T三种字母。
输出格式
在一行中输出给定字符串中包含多少个PAT。由于结果可能比较大,只输出对1000000007取余数的结果。
输入样例
APPAPT
输出样例
2
思路
sum of 每个’A’ 左边’P’个数 * 右边’T’个数
代码
/**
* @tag PAT_B_1040
* @authors R11happy (xushuai100@126.com)
* @date 2016-9-7 17:58-18:20
* @version 1.0
* @Language C++
* @Ranking 765/583
* @function null
*/
//sum of 每个'A' 左边'P'个数 * 右边'T'个数
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cstring>
using namespace std;
const int maxn = 100000;
char str[maxn];
const long long MOD = 1000000007;
int main(int argc, char const *argv[])
{
long long sum_T = 0L;
long long left_P = 0L, right_T = 0L;
long long ans = 0L;
gets(str);
int len = strlen(str);
//统计总的'T'个数
for(int i = 0; i<len; i++)
{
if(str[i] == 'T') sum_T++;
}
for(int i = 0; i<len; i++)
{
if(str[i] == 'P') left_P++; //'A'左边'P'的个数
else if(str[i] == 'A')
{
ans += left_P * (sum_T - right_T) % MOD;
ans = ans % MOD;
}
else right_T++; //'A'左边'T'的个数
}
printf("%lld\n",ans );
return 0;
}