HDU 5198 /BC 36A Strange Class

1.判断长度能否除3

2判断每一段内部是否相同

3不同段之间不能相同

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cctype>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<algorithm>
#include<set>
#define scnaf scanf
#define cahr char
#define bug puts("bugbugbug");
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
const int mod=1000000007;
const int maxn=1e6+100;
const int inf=1e9;
char a[100];
bool pd()
{
    int la=strlen(a);
    if(la%3) return 0;
    int la3=la/3;
    for(int i=1;i<la3;i++)
        if(a[i]!=a[i-1])return 0;
    for(int i=la3+1;i<2*la3;i++)
        if(a[i]!=a[i-1])return 0;
    for(int i=2*la3+1;i<la;i++)
        if(a[i]!=a[i-1])return 0;
        if(a[0]==a[la3]||a[2*la3]==a[la3]||a[2*la3]==a[la])
            return 0;
        return 1;

}
int main()
{
    while(~scanf("%s",a))
    {
        if(pd())puts("YES");
        else puts("NO");
    }
}