1006.ZOJ问题

题目描述：
对给定的字符串(只包含'z','o','j'三种字符),判断他是否能AC。

是否AC的规则如下：
1. zoj能AC；
2. 若字符串形式为xzojx，则也能AC，其中x可以是N个'o' 或者为空；
3. 若azbjc 能AC，则azbojac也能AC，其中a,b,c为N个'o'或者为空；
输入：
输入包含多组测试用例，每行有一个只包含'z','o','j'三种字符的字符串，字符串长度小于等于1000。
输出：
对于给定的字符串，如果能AC则请输出字符串“Accepted”，否则请输出“Wrong Answer”。
样例输入：
zoj
ozojo
ozoojoo
oozoojoooo
zooj
ozojo
oooozojo
zojoooo
样例输出：
Accepted
Accepted
Accepted
Accepted
Accepted
Accepted
Wrong Answer

Wrong Answer

#include <iostream>
#include <string>
#include <algorithm>

using namespace std;

//
//不仅要满足（z之前的o个数n1，z和j之间的o个数为n2，j后面的o个数为n3）n1*n2=n3的关系
//还需要一些特殊判断，比如z或j的个数不为1，或者j在z前面，均输出WA

int main()
{
	string str;
	while(cin  >> str)
	{
		int number_z = count(str.begin(),str.end(),'z');
		int number_j = count(str.begin(),str.end(),'j');
		if(!(number_z == 1 && number_j == 1))
		{
			cout << "Wrong Answer" << endl;
			continue;                          //这里不是break,break跳出整个循环了
		}
		string::size_type pos1 = str.find('z');
		string::size_type pos2 = str.find('j');
		if(pos1 > pos2)
		{
			cout << "Wrong Answer" << endl;
			continue;
		}
		string str1 = str.substr(0,pos1);
		string str2 = str.substr(pos1 + 1,pos2 - pos1 - 1);
		string str3 = str.substr(pos2 + 1);
		int n1 = count(str1.begin(),str1.end(),'o');
		int n2 = count(str2.begin(),str2.end(),'o');
		int n3 = count(str3.begin(),str3.end(),'o');
		if(n2 != 0 && n3 / n2 == n1)
			cout << "Accepted" << endl;
		else
			cout << "Wrong Answer" << endl;
	}
	return 0;
}