[프로그래머스] 가장 큰 정사각형 찾기

https://programmers.co.kr/learn/courses/30/lessons/12905

코딩테스트 연습 - 가장 큰 정사각형 찾기

 

문제 설명

1와 0로 채워진 표(board)가 있습니다. 표 1칸은 1 x 1 의 정사각형으로 이루어져 있습니다. 표에서 1로 이루어진 가장 큰 정사각형을 찾아 넓이를 return 하는 solution 함수를 완성해 주세요. (단, 정사각형이란 축에 평행한 정사각형을 말합니다.)

예를 들어

1234

0	1	1	1
1	1	1	1
1	1	1	1
0	0	1	0

가 있다면 가장 큰 정사각형은

1234

0	1	1	1
1	1	1	1
1	1	1	1
0	0	1	0

가 되며 넓이는 9가 되므로 9를 반환해 주면 됩니다.

제한사항

• 표(board)는 2차원 배열로 주어집니다.
• 표(board)의 행(row)의 크기 : 1,000 이하의 자연수
• 표(board)의 열(column)의 크기 : 1,000 이하의 자연수
• 표(board)의 값은 1또는 0으로만 이루어져 있습니다.

---

입출력 예

boardanswer

[[0,1,1,1],[1,1,1,1],[1,1,1,1],[0,0,1,0]]	9
[[0,0,1,1],[1,1,1,1]]	4

입출력 예 설명

입출력 예 #1
위의 예시와 같습니다.

입출력 예 #2
| 0 | 0 | 1 | 1 |
| 1 | 1 | 1 | 1 |
로 가장 큰 정사각형의 넓이는 4가 되므로 4를 return합니다.

 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int arr[1001][1001];
int w, h;

// x,y 위치에 사각형 확인
int square(int x, int y) {
    int current = arr[y][x];
    if (current != 0) {
        int left = arr[y][x - 1];
        int up = arr[y - 1][x];
        int up_left = arr[y - 1][x - 1];

        arr[y][x] = min(min(left, up), up_left) + 1;
    }
    return arr[y][x];
}

int solution(vector<vector<int>> board)
{
    int answer = 0;
    w = board[0].size();
    h = board.size();

    if (w <= 1 || h <= 1)
        return 1;

    for (int i = 0; i < h; i++) {
        for (int j = 0; j < w; j++) {
            arr[i][j] = board[i][j];
        }
    }

    for (int i = 1; i < h; i++) {
        for (int j = 1; j < w; j++) {
            answer = max(answer,square(j, i));
        }
    }

    return answer * answer;
}

핵심은

 

현재 위치에서 만들 수 있는 가장 큰 정사각형 = 근처에 사각형 크기[(왼쪽, 위, 왼쪽위) 혹은 (오른쪽, 아래, 오른쪽아래)] 값중에 가장 작은 값에 1을 더한값

 

arr[y][x] = min(arr[y-1][x], arr[y][x-1], arr[y-1][x-1]) + 1

 

입니다.