【CodeForces】837B - Flag of Berland

最新推荐文章于 2021-06-14 22:54:33 发布

原创最新推荐文章于 2021-06-14 22:54:33 发布 · 400 阅读

0 ·

CC 4.0 BY-SA版权

———数学——— 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一个用于判断由三种颜色组成的旗帜图案是否相同的算法。通过分析旗帜的横纵比，检查颜色分布的一致性来实现这一目标。算法首先验证旗帜是否可以被均匀划分为三等份，并确保每种颜色出现的次数相等。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

http://codeforces.com/problemset/problem/837/B

一面旗帜上有3种颜色，判断三种颜色构成的图形形状是否相同。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <map>
#include <algorithm>
using namespace std;

int n,m;
char g[105][105];
int cnt[300]={0};
bool flag,ok;


int main(){

    cin >> n >> m;
    for (int i=0;i<n;i++){
        for (int j=0;j<m;j++){
            cin >> g[i][j];
        }
    }

    if (n%3==0){
        flag=true;
        for (int i=0;i<n;i++){
            for (int j=0;j<m;j++){
                if (g[i][j]!=g[i/(n/3)*(n/3)][0]){
                    flag=false;
                }
                cnt[g[i][j]]++;
            }
        }
        if (flag){
            ok=true;
        }
    }

    if (m%3==0){
        flag=true;
        for (int i=0;i<n;i++){
            for (int j=0;j<m;j++){
                if (g[i][j]!=g[0][(j/(m/3))*(m/3)]){
                    flag=false;
                }
                cnt[g[i][j]]++;
            }
        }
        if (flag){
            ok=true;
        }
    }

    if (ok&&(cnt['R']==cnt['G']&&cnt['R']==cnt['B'])){
        cout << "YES" << endl;
    }
    else {
        cout << "NO" << endl;
    }
}