PAT.1003. 我要通过！(20) C++实现

#include<iostream>
#include<string.h>

using namespace std;



int judge()
{
  char number[100];
  cin >> number;
  int a, b, c;
  int p = 0, q = 0 ;
  bool result;
  for (int i = 0; i < strlen(number); i++)
  {
     if(  number [i] != 'P'&& number[i]!='A'&& number[i]!='T')
    {
       result = 0;    
       break;
    }
    else if ( number[i] == 'P')
    {
      p = i;
    }
    else if ( number[i] == 'T')
    {
      q = i;
    }

  }
  a = p;
  b = q - p - 1;
  c = strlen(number) - 1 - q;

  if (a*b == c)
  {
    result = 1;
  }
  if (b == 0)
  {
    result = 0;

  }
  return result;


}

void show(bool result)
{
  if (result==1)
    cout << "YES" << endl;
  else
    cout << "NO" << endl;
}
int main()
{
  int n;
  cin >> n;
  char all[10][100];
  bool result[10];
  for (int i = 0; i < n; i++)
  {
    result[i]=judge() ;
  }
  for (int j = 0; j < n; j++)
  {
    show(result[j]);
  }
  return 0;
}

小白我本人，so我的算法都是很简单的了 一目了然哈哈