1040
字符串 APPAPT 中包含了两个单词 PAT,其中第一个 PAT 是第 2 位§,第 4 位(A),第 6 位(T);第二个 PAT 是第 3 位§,第 4 位(A),第 6 位(T)。
现给定字符串,问一共可以形成多少个 PAT?
输入格式:
输入只有一行,包含一个字符串,长度不超过10^
5
,只包含 P、A、T 三种字母。
输出格式:
在一行中输出给定字符串中包含多少个 PAT。由于结果可能比较大,只输出对 1000000007 取余数的结果。
输入样例:
APPAPT
输出样例:
2
思路:我的思路是先得出T的总个数,然后判断遇到P的个数,之后每遇到一次T,之前T的次数就减少一个,然后遇到A就得到当前的总PAT个数
package test1;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class PTA1040 {
public static void main(String[] args) throws IOException {
BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
char[] s = in.readLine().toCharArray();
in.close();
int countP = 0, countT = 0;
long pat = 0;
for (char c : s) {
if (c == 'T')
countT++;
}
for (char c : s) {
if (c == 'P')
countP++;
if (c == 'T')
countT--;
if (c == 'A') {
pat = pat + countP * countT;
if (pat > 1000000007)
pat = pat % 1000000007;
}
}
System.out.println(pat);
}
}
运行结果:
PATPPTAPTAPATPATPATATPTATATATAAPTAPTATP
524