Educational Codeforces Round 26 B - Flag of Berland【思维】

题意：判断一个矩阵 ， 是否满足以下条件：
1.仅有R、G、B三种字符
2.三种字符各自连一片，分别和矩形的边平行
3.RGB三种字符各自构成的矩形中，长宽要相等，并且平行

思路：对n和m是不是3的倍数展开讨论，如果n是3的倍数，那么分成[1,n/3] [n/3+1,2*n/3] [2*n/3+1,n]三个区间进行讨论。m同理。

复杂度分析： 图也就100*100 只要不写成死循环就没问题

#include <bits/stdc++.h>
using namespace std;
char mp[200][200];
int main(void)
{
    int n,m;
    cin >> n >>m;
    for(int i=1;i<=n;i++)
        scanf("%s",mp[i]+1);
    int a=0,b=0,c=0;
    for(int i=1;i<=n;i++)//判断是否有R/G/B;
    {
        for(int j=1;j<=m;j++)
        {
            if(mp[i][j]=='R')
                a++;
            else if(mp[i][j]=='G')
                b++;
            else
                c++;
        }
    }
    if(a*b*c==0)
    {
        printf("NO\n");
        return 0;
    }
    int flag1=1,flag2=1;
    if(n%3!=0 && m%3!=0) //至少有一个要是3的倍数
    {
        printf("NO\n");
        return 0;
    }
    if(n%3==0)//枚举区间
    {
        for(int i=1;i<=n/3;i++)
        {
            for(int j=1;j<=m;j++)
            {
                if(mp[i][j]!=mp[1][1])
                {
                    flag1=0;
                }
            }
        }
        for(int i=n/3+1;i<=2*n/3;i++)
        {
            for(int j=1;j<=m;j++)
            {
                if(mp[i][j]!=mp[n/3+1][1])
                {
                    flag1=0;
                }
            }
        }
        for(int i=2*n/3+1;i<=n;i++)
        {
            for(int j=1;j<=m;j++)
            {
                if(mp[i][j]!=mp[2*n/3+1][1])
                {
                    flag1=0;
                }
            }
        }
        if(flag1)   {printf("YES\n");   return 0;}
    }
    if(m%3==0)
    {
        //cout << "woc" << endl;
        for(int j=1;j<=m/3;j++)
        {
            for(int i=1;i<=n;i++)
            {
                if(mp[i][j]!=mp[1][1])
                {
                    flag2=0;
                }
            }
        }
        for(int j=m/3+1;j<=2*m/3;j++)
        {
            for(int i=1;i<=n;i++)
            {
                if(mp[i][j]!=mp[1][m/3+1])
                {
                    flag2=0;
                }
            }
        }
        for(int j=2*m/3+1;j<=m;j++)
        {
            for(int i=1;i<=n;i++)
            {
                if(mp[i][j]!=mp[1][2*m/3+1])
                {
                    flag2=0;
                }
            }
        }
         if(flag2)   {printf("YES\n");   return 0;}
    }
    printf("NO\n");
    return 0;
}