HDU 5198 水 c语言

#include<stdio.h>
#include<string.h>
int main()
{
char arr[1000];
int i,size,flag,num;
while(scanf("%s",arr) != EOF)
{
size=strlen(arr);flag=0;num=1;
if(size%3==0)
{
size=size/3;
for(i=0; arr[i] != '\0';i++)
{
if(arr[i]==arr[i+1])
num++;
else
{
if(num==size){
flag++;
}
num=1;
}
}
if(flag==3) printf("YES\n");
else printf("NO\n");
}
else
{
printf("NO\n");
}
}

return 0;
}