프로그래머스 - 가장 큰 정사각형 찾기 - C++
문제 난이도
level 2
문제 설명
1와 0로 채워진 표(board)가 있습니다. 표 1칸은 1 x 1 의 정사각형으로 이루어져 있습니다. 표에서 1로 이루어진 가장 큰 정사각형을 찾아 넓이를 return 하는 solution 함수를 완성해 주세요. (단, 정사각형이란 축에 평행한 정사각형을 말합니다.)
제한사항
- 표(board)는 2차원 배열로 주어집니다.
- 표(board)의 행(row)의 크기 : 1,000 이하의 자연수
- 표(board)의 열(column)의 크기 : 1,000 이하의 자연수
- 표(board)의 값은 1또는 0으로만 이루어져 있습니다.
문제 풀이
이 문제는 DP를 이용하여 해결하였다. 알고리즘을 간단히 설명하자면 board[1][1]부터 반복을 시작하여 위, 왼쪽, 왼쪽 대각선과 board[1][1]을 각각 비교하여 (가장 작은 수 + 1)을 board[1][1]의 값으로 바꾸어준다. 이때의 board[1][1] 값이 바로 1로된 정사각형의 한 변의 길이가 된다. 이 일련의 과정을 반복하면서 한 변의 길이의 최댓값을 answer에 저장해 두면 answer가 1로된 정사각형의 한 변의 최대 길이가 되어 최대 넓이를 구할 수 있게된다.
핵심 코드는 이부분이다.
board[i][j] = min({board[i-1][j-1], board[i-1][j] , board[i][j-1]}
풀이 코드
#include <vector>
#include <algorithm>
using namespace std;
int solution(vector<vector<int>> board)
{
int answer = 0;
if(board.size()<2 || board[0].size()<2){
return 1;
}
for(int i=1;i<board.size();++i)
for(int j=1;j<board[0].size();++j)
if(board[i][j]==1){
board[i][j] = min({board[i-1][j-1], board[i-1][j] , board[i][j-1]})+1;
answer = max(answer,board[i][j]);
}
return answer*answer;
}