<백준 알고리즘> 1051번: 숫자 정사각형

문제

N*M크기의 직사각형이 있다. 각 칸은 한 자리 숫자가 적혀 있다. 이 직사각형에서 꼭짓점에 쓰여 있는 수가 모두 같은 가장 큰 정사각형을 찾는 프로그램을 작성하시오. 이때, 정사각형은 행 또는 열에 평행해야 한다.

입력

첫째 줄에 N과 M이 주어진다. N과 M은 50보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에 수가 주어진다.

출력

첫째 줄에 정답 정사각형의 크기를 출력한다.

예제 입력 1

3 5 42101 22100 22101

예제 출력 1

9

---

Solution

#include<stdio.h>
#include<stdlib.h>

int main(void)
{
	int N, M, **rectangle=NULL, maxmax=0;

	scanf("%d %d", &N, &M);
	rectangle=(int **)malloc(N*sizeof(int *));
	for(int n=0;n<N;n++)
		rectangle[n]=(int *)malloc(M*sizeof(int));

	for(int n=0;n<N;n++)
		for(int m=0;m<M;m++)
			scanf("%1d", &rectangle[n][m]);

	for(int n=0;n<N;n++)
		for(int m=0,max=0;m<M;m++,max=0)
			while(n+max<N && m+max<M)
			{
				if(rectangle[n][m]==rectangle[n][m+max] && rectangle[n][m]==rectangle[n+max][m] && rectangle[n][m]==rectangle[n+max][m+max] && max>maxmax)
					maxmax=max;
				max++;
			}

	printf("%d\n", (maxmax+1)*(maxmax+1));
	for(int n=0;n<N;n++)
		free(rectangle[n]);
	free(rectangle);
	return 0;
}