這道題思路就是 用A之前的P*A之後的T就是總的PAT個數
題目鏈接:https://www.patest.cn/contests/pat-b-practise/1040
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
#include<iostream>
#include<cstdio>
#include<string>
using namespace std;
int main()
{
string s;
cin>>s;
int count_T=0,count_P=0;
for(int i=0;i<s.length();i++)
{
if(s[i]=='T')
count_T++;
}
long long total=0;
for(int i=0;i<s.size();i++)
{
switch(s[i])
{
case 'P':
count_P++;
break;
case 'T':
count_T--;
break;
case 'A':
total+=count_P*count_T;
break;
}
}
total = total%1000000007;
printf("%lld", total);
return 0;
}