暴走的字符串

链接：https://ac.nowcoder.com/acm/problem/26306
来源：牛客网

样例1
输入：

ABCD
ABCDABCD
ABCDABCDABCD
AAAA
BBBBBBBB
CCCCCCCCCCCC

输出：

NO
YES

AC代码

#include<bits/stdc++.h>
using namespace std;
int a[300];
int main() {
	string s1,s2,s3;
	while(cin>>s1>>s2>>s3) {
		int flag=1;
		for(int i=0; i<4 && flag; i++) {
			memset(a,0,sizeof(a));//初始化 
			a[s1[i]]++;
			a[s2[i]]++;
			a[s2[i+4]]++;
			a[s3[i]]++;
			a[s3[i+4]]++;
		    a[s3[i+8]]++; 
			if(!(a['A']== 1 && a['B']== 2 && a['C']== 3))//NO
			{
			cout<<"NO"<<endl; 
			flag=0;
		    }
		}
		if(flag)
		cout<<"YES"<<endl;
	}
	return 0;
}