1093 Count PAT’s (25 分)
-
题目大意:找出给出字符串里包含"PAT"的次数
最后要将结果与1000000007除余(一开始没看懂,原来mode是除余的意思)
-
思路1(用了三重循环,超时了):依次遍历每个字符(i >= len(str)截止)
-
思路2(只需一个循环):对一个A而言,它构成的"PAT"个数为它左边的P的个数和右边的T的个数相乘。总的"PAT"个数为所有A构成的"PAT"个数之和。
-
先统计一共有多少个T:
cntt
-
按顺序遍历字符串,若:
-
s[i] == 'P':cntp++
-
s[i] == 'T':cntt--
-
s[i] == 'A':result = (result + cntp * cntt) % 1000000007;
因为测试样例结果过大,需要将结果取模,但是如果是在最后取模,因为在中间过程已经越界,结果还是不对的,导致最后两个测试点过不去。正确的做法应该是在遍历每个A的时候就取模,有效防止越界(参考自)
-
-
-
知识点:
- 取余操作:循环相乘/相加后再取余和每次循环都取余的结果一样。
-
代码:
#include <iostream> #include <string> using namespace std; int main() { string s; cin >> s; int cntp = 0, cntt = 0, result = 0; for(int i = 0; i < s.length(); i++) if(s[i] == 'T') cntt++; for(int i = 0; i < s.length(); i++){ if(s[i] == 'P') cntp++; if(s[i] == 'T') cntt--; if(s[i] == 'A') result = (result + cntp * cntt) % 1000000007; } printf("%d", result); return 0; }
-
总结:
- 取余操作:循环相乘/相加后再取余和每次循环都取余的结果一样。
- 做题要多动脑,找规律。