PAT Basic1040 有幾個PAT
阿新 • • 發佈:2018-12-12
字串 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
挺巧妙的一道題。
要組成PAT的只要相對位置是 P … A … T 就行。
所以對於1個T,能組成多少PAT取決於前面有多少對P… A;
而對於1個A,能組成多少PA取決於前面有多少P。
這樣問題就解決了。
以下程式碼:
#include<iostream>
#include<cstdio>
using namespace std;
int main()
{
char ch;
long long P=0,PA=0,PAT=0;
while((ch=getchar())!='\n')
{
if(ch=='P')
P++;
else if(ch=='A')
PA+=P;
else if(ch=='T')
PAT+=PA;
PAT%=1000000007; //記得每次求模
}
cout<<PAT<<endl;
return 0;
}