洛谷 P3392 涂国旗

题目描述
某国法律规定，只要一个由N*M个小方块组成的旗帜符合如下规则，就是合法的国旗。（毛熊：阿嚏——）
从最上方若干行（>=1）的格子全部是白色的。
接下来若干行（>=1）的格子全部是蓝色的
剩下的行（>=1）全部是红色的
现有一个棋盘状的破布，分成了N行M列的格子，每个格子是白色蓝色红色之一，小a希望把这个布改成该国国旗，方法是在一些格子上涂颜料，盖住之前的颜色。
小a很懒，希望涂最少的格子，使这块破布成为一个合法的国旗。
输入输出格式
输入格式：

第一行是两个整数，N,M
接下来N行是一个矩阵，矩阵的每一个小方块是’W’（白）,’B’（蓝）,’R’（红）中的一个

输出格式：

一个整数，表示至少需要涂多少块。

输入输出样例
输入样例#1：
4 5
WRWRW
BWRWB
WRWRW
RWBWR
输出样例#1：
11
说明
样例解释：
目标状态是
WWWWW
BBBBB
RRRRR
RRRRR
一共需要改11个格子
对于100%的数据，N，M<=50

---

先计算出每行改为，其他颜色需要改动的块数。
由于行都是连续的，想到前缀和优化。
枚举中间颜色，计算最小。

---

#include<iostream>#include<cstdio>using namespace std;int n,m,mp[101][101],c[101][5],ans=1e9+7;int main(){    scanf("%d%d",&n,&m);    char ch[101];    for(int i=1;i<=n;i++)    {        scanf("%s",ch+1);        for(int j=1;j<=m;j++)        {            if(ch[j]=='W')                mp[i][j]=1;            else if(ch[j]=='B')                mp[i][j]=2;            else                mp[i][j]=3;        }    }    for(int i=1;i<=n;i++)        for(int j=1;j<=3;j++)        {            for(int k=1;k<=m;k++)                if(mp[i][k]!=j)                    c[i][j]++;            c[i][j]+=c[i-1][j];        }    for(int i=2;i<=n-1;i++)//枚举中间         for(int j=i;j<=n-1;j++)        {            int tmp=c[i-1][1]+c[j][2]-c[i-1][2]+c[n][3]-c[j][3];            ans=min(ans,tmp);        }    printf("%d\n",ans);    return 0;}