[백준]#1051 숫자 정사각형

문제

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

입력

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

출력

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

예제 입력 1

3 5
42101
22100
22101

예제 출력 1

9

풀이

전형적인 브루트 포스 문제이다. 그러나 출력 조건에 default 출력이 1이라는 문항이 없어서 첫 시도에서 틀렸다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] str = br.readLine().split(" ");
        int N = Integer.parseInt(str[0]);
        int M = Integer.parseInt(str[1]);
        int[][] arr = new int[N][M];

        for(int i=0; i<N; i++) {
            String input = br.readLine();
            for(int j=0; j<M; j++) {
                arr[i][j] = input.charAt(j) - '0';
            }
        }
        int length = 2;
        int max = 1;

        while(length<=M && length<=N) {
            for(int i=0; i+length-1<N; i++) {
                for(int j=0; j+length-1<M; j++) {
                    if(arr[i][j]==arr[i+length-1][j] && arr[i+length-1][j]==arr[i][j+length-1] && arr[i][j+length-1]==arr[i+length-1][j+length-1]) {
                        if(max<length)
                            max=length;
                    }
                }
            }
            length++;
        }

        System.out.println(max*max);
    }
}