PAT 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
能力不够,第一次写三个测试点运行超时,优化后还是不行,后来参考http://blog.csdn.net/ice_camel/article/details/44263451的思路才运行成功。
1 #include<stdio.h> 2 #include<string.h> 3 #include<stdlib.h> 4 #include<ctype.h> 5 #include<math.h> 6 int main(){ 7 char a[100010]; 8 gets(a); 9 int lena = strlen(a); 10 int t=0,at=0,pat=0; 11 for(int i=lena-1;i>=0;i--){ 12 if(a[i]=='T'){ 13 t++; 14 }else if(a[i]=='A'){ 15 at = at+t; 16 }else if(a[i]=='P'){ 17 pat = (pat+at)%1000000007; 18 } 19 } 20 printf("%d",pat); 21 }