HDU--3788--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'或者为空；

Input

输入包含多组测试用例，每行有一个只包含'z','o','j'三种字符的字符串，字符串长度小于等于1000；

Output

对于给定的字符串，如果能AC则请输出字符串“Accepted”，否则请输出“Wrong Answer”。

Sample Input
zoj
ozojo
ozoojoo
oozoojoooo
zooj
ozojo
oooozojo
zojoooo
Sample Output
Accepted
Accepted
Accepted
Accepted
Accepted
Accepted
Wrong Answer
Wrong Answer

思路：前面的o的个数乘以中间o的个数等于后面o的个数，并保证o的个数不为0且"z","j"的个数只有一个；

代码如下：

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxa=1e3+10;
char a[maxa];
bool check(char *a){
	int cntl=0,cntm=0,cntr=0,cnt0=0;//统计o的个数，z左边，z与j中间，j右边; 
	int z=0,j=0;
	int tempz=0,tempj=0;
	for(int i=0;i<strlen(a);i++){
		if(a[i]=='z')	z++,tempz=i;
		if(a[i]=='j')	j++,tempj=i;
		if(a[i]=='o')	cnt0++;
	}
	if(z!=1||j!=1||!cnt0)	return false;
	cntl=tempz,cntm=tempj-tempz-1,cntr=strlen(a)-tempj-1;
	//printf("#%d #%d #%d\n",cntl,cntm,cntr);
	if(cntl*cntm==cntr)	return true;
	return false;
}
int main(){
	while(scanf("%s",a)!=EOF){
		if(check(a))	printf("Accepted\n");
		else	printf("Wrong Answer\n");
	}
	return 0;
}