洛谷 #2733. 家的范围

题意

在一个边长为n的正方形矩阵中，寻找边长为2~n的全1矩阵的个数

题解

按边长找，每次的g[i][j]（bool）表示当前边长下起点为i,j的矩阵是否符合要求

若g[i][j]、g[i + 1][j]、g[i][j + 1]、g[i + 1][j + 1]均为true，则新的g[i][j]也为true，否则为false

我自己想出来的O(\(n^3\))算法

调试记录

无

#include <cstdio>
#include <cstring>
#define maxn 255

using namespace std;

bool pre[maxn][maxn], next[maxn][maxn];
int n, cnt;

int main(){
	scanf("%d", &n);
	
	char str[maxn];
	for (int i = 1; i <= n; i++){
		scanf("%s", str);
		for (int j = 0; j < n; j++) pre[i][j + 1] = str[j] - '0';
	}
	
	for (int len = 2; len <= n; len++){
		cnt = 0;
		for (int x = 1; x <= n - len + 1; x++){
			for (int y = 1; y <= n - len + 1; y++){
				if (pre[x][y] && pre[x][y + 1] && pre[x + 1][y] && pre[x + 1][y + 1]) next[x][y] = true, cnt++;
			}
		}
		
		if (!cnt) return 0;
		//memcpy(pre, next, maxn * maxn + 1);  
		for (int i = 1; i <= n; i++) for (int j = 1; j <= n; j++) pre[i][j] = next[i][j], next[i][j] = 0;
		printf("%d %d\n", len, cnt);
	}
	
	return 0;
}