P3392 涂条纹

题目描述

只要一个由 $N\times M$ 个小方块组成的旗帜符合如下规则，就是合法的图案。

• 从最上方若干行（至少一行）的格子全部是白色的；
• 接下来若干行（至少一行）的格子全部是蓝色的；
• 剩下的行（至少一行）全部是红色的；

现有一个棋盘状的布，分成了 $N$ 行 $M$ 列的格子，每个格子是白色蓝色红色之一，小 a 希望把这个布改成合法图案，方法是在一些格子上涂颜料，盖住之前的颜色。

小 A 很懒，希望涂最少的格子，使这块布成为一个合法的图案。

输入格式

第一行是两个整数 $N,M$。

接下来 $N$ 行是一个矩阵，矩阵的每一个小方块是 W（白），B（蓝），R（红）中的一个。

输出格式

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

样例 #1

样例输入 #1

4 5
WRWRW
BWRWB
WRWRW
RWBWR

样例输出 #1

11

提示

样例解释

目标状态是：

WWWWW
BBBBB
RRRRR
RRRRR

一共需要改 $11$ 个格子。

数据范围

对于 $100\%$ 的数据，$N,M\le 50$。

思路

前缀和 + 枚举。具体见代码注释。
参考大佬题解。

代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
#define endl '\n'
using namespace std;

const int N = 60;

int n, m;
vector<string> g;
int w[N], b[N], r[N];
// 分别代表需要将前i行全部变为蓝色、白色、红色，所需要涂的块数，即前缀和
// 这样就可以在之后枚举蓝白和白红分界点的时候，快速求出索要涂的块数

int main() {
	ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
	cin >> n >> m;
	for (int i = 1; i <= n; i++) {
		string s;
		cin >> s;
		g.push_back(s);

		for (int j = 0; j < s.size(); j++) {
			// 如果是某种颜色，将该颜色涂成另外两种颜色的个数就要增加
			if (s[j] == 'W') {
				b[i]++;
				r[i]++;
			}
			else if (s[j] == 'B') {
				w[i]++;
				r[i]++;
			}
			else {
				w[i]++;
				b[i]++;
			}
		}

		w[i] += w[i - 1];
		b[i] += b[i - 1];
		r[i] += r[i - 1];
	}

	int ans = 10000, sum;
	int wt, bt, rt;// 三种颜色所需要涂的个数

	// 枚举分界点
	for (int i = 1; i <= n - 2; i++) {// 蓝白分界点
		for (int j = i + 1; j <= n - 1; j++) {// 白红分界点
			wt = w[i];
			bt = b[j] - b[i];
			rt = r[n] - r[j];
			sum = wt + bt + rt;

			ans = min(ans, sum);
		}
	}

	cout << ans << endl;
	return 0;
}