P1191 矩形

题目描述

给出一个n×n的矩阵，矩阵中，有些格子被染成白色，有些格子被染成黑色，现要求矩阵中白色矩形的数量

输入格式

第一行，一个整数n，表示矩形的大小。

接下来n行，每行n个字符，这些字符为“W”或“B”。其中“W”表示白格，“B”表示黑格。

输出格式

一个正整数，为白色矩形数量

输入输出样例

输入 #1

4
WWBW
BBWB
WBWW
WBWB

输出 #1

15

说明/提示

对于30%的数据，n≤50；

对于100%的数据，n≤150；

思路

矩形固定了左下角向右上方拓展着去找，即不会重复，也使思维变得简单了很多

每行开始时，预处理这一行每一列的高度，方便之后的统计， 然后确认往右拓展，能取的高度是拓展到的高度中的最小值，然后会发现以i, j为左下角宽度为k-j+1的矩形数量在数值上等于高度，累加后就是答案了

代码

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

const int N=200;

char map[N][N];
int n,ans,cnt[N];

int main () {
	scanf("%d",&n);
	for(int i=1; i<=n; i++)
		for(int j=1; j<=n; j++)
			cin>>map[i][j];
	for(int i=1; i<=n; i++) {
		for(int j=1; j<=n; j++)
			if(map[i][j]=='B')
				cnt[j]=0;
			else
				cnt[j]++;
		for(int j=1; j<=n; j++) {
			int flag=cnt[j];
			for(int k=j; k<=n; k++) {
				if(!cnt[k])
					break;
				flag=min(flag,cnt[k]);
				ans+=flag;
			}
		}
	}
	printf("%d\n",ans);
	return 0;
}