PTA 1003 我要通过！

1003 我要通过！

解析看了一篇比较好的，下面有题意解析，直接摘过来了

一开始没有读懂题意，一直在想怎么将字符串分词解决，后来百度了一下题解，才发现这是一个完完全全的数学问题。

解决问题的关键是第二个和第三个要求。

（1）第二个要求说明了a的长度和c的长度是相等的，所以我们可以列出一个等式：lena = lenc；

（2）第三个要求则说明，在第二个条件成立的条件下，将aPATc中的A字符换成b串，变为aPbTc。如果该字符串正确，那么aPbATac字符串也正确。仔细观察不难发现，P和A字符中间多加了一个A字符，而字符串末尾c串后面则加了一个a串。也就是说，c串后面加的a串的个数与P字符和T字符中间加入的A字符的个数相等。

假设b含有lenb个A字符，那么新加入的A字符的个数就是（lenb-1）个(因为为了保证字符串正确，b中最起码要有一个A字符)。所以c串后面的a串个数也就是（lenb-1）个。所以可以列出一个式子： lenc = lenc+（lenb-1）*lena 。

由第二个要求得出的式子 lena = lenc 带入得： lenc = lena + （lenb-1）*lena 。

化简上式即为 ： lenc = lena * lenb。
————————————————
版权声明：本文为CSDN博主「二木mua」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/ermua/article/details/79513355

下面是ac代码

#include<iostream>
using namespace std;
string s;

bool vis(){
    int cnt = 0;
	for(int i = 0;i < (int)s.size();i++)
	{
        if(s[i] == 'P') cnt++;
        else if(s[i] == 'T') cnt++;
		if(s[i] != 'P' && s[i] != 'A' && s[i] != 'T') return false;
	}	
	return cnt == 2;//统计P和T只存在两个的情况,排除PT多存在的情况
}

int main() {
	int n;cin >> n;
	getchar();
	for(int k = 0;k < n;k++)
	{
		getline(cin,s);
		if(vis()) {
			int i = 0, la = 0, lb = 0, lc = 0;
			while(s[i++] != 'P' && i < (int)s.size()) la++;//统计a对应部分A字符个数
			i++;
			while(s[i++] != 'T' && i < (int)s.size()) lb++;//统计b对应部分B字符个数，注意需要-2，因为题目给的aPbATca中b部分应当减去一个A数量
			lb = i - la - 2;
			i++;
			lc = (int)s.size() - i + 1;//最后部分对应的数量可以直接用字符长度简便计算
			if(la * lb == lc) puts("YES");
			else puts("NO");
		}else {
			puts("NO");	
		}
	}
	return 0;
}