PAT-1003 我要通过！ (20分) JavaScript(node)

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 PAT 的“答案正确”大派送 —— 只要读入的字符串满足下列条件，系统就输出“答案正确”，否则输出“答案错误”。

得到“答案正确”的条件是：

字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
任意形如 xPATx的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A组成的字符串；
如果aPbTc是正确的，那么 aPbATca 也是正确的，其中 a、 b、c均或者是空字符串，或者是仅由字母A 组成的字符串。

现在就请你为 PAT 写一个自动裁判程序，判定哪些字符串是可以获得“答案正确”的。

输入格式：

每个测试输入包含 1 个测试用例。第 1 行给出一个正整数 n (<10)，是需要检测的字符串个数。接下来每个字符串占一行，字符串长度不超过 100，且不包含空格。

输出格式：

每个字符串的检测结果占一行，如果该字符串可以获得“答案正确”，则输出 YES，否则输出 NO。

输入样例：

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO

解题思路:

通过读取题干信息，字符串中必须仅有 P、 A、 T、及空白字符，且有且仅有一个P,T, 且P在T之前；
还有一个隐藏的规律，以P和T为边界可以分为三部分： ********** P*********** T********，其中 全部由空白符和A组成；
设：A-P之间*个数：a， P-T之间*个数:b ,T-A之间*个数:c，得：a*b=c ;
将全部*****替换为空时，那么用例 == 'PT'时，有可能会YES, 否则直接判定NO;
通过字符串的截取，求出上述变量a,b,c,判定上述条件是否成立，返回结果；

JavaScript(node 12.18.4)

var buf = "";
process.stdin.on("readable", function () {
    var chunk = process.stdin.read();
    if (chunk) buf += chunk.toString();
});

process.stdin.on("end", function () {
    const a = compute(buf);
    console.log(a);
});
function compute(num) {
  var tmp = num.split('
');
  num = tmp.slice(1,tmp[0]-0 + 1);
  var result = '';
  for(var i=0;i<num.length;i++){
      var tmp1 = num[i].replace(/A|s/g,'');
      var tmp2 = num[i];
      if(num[i].length>=3 && tmp1.length == 2 && tmp1[0]=='P' && tmp1[1] == 'T'){
        var $1 = tmp2.substring(0,tmp2.indexOf('P')).length;
        var $2 = tmp2.substring(tmp2.indexOf('P')+1,tmp2.indexOf('T')).length;
        var $3 = tmp2.substring(tmp2.indexOf('T')+1).length;
        if($1*$2 == $3){
          result+='YES';
        }else{
          result+='NO';  
        }
      }else{
         result+='NO';
      }
      result+= i == num.length-1 ? '' : '
';
    }
    return result;
}