1와 0로 채워진 표(board)가 있습니다. 표 1칸은 1 x 1 의 정사각형으로 이루어져 있습니다. 표에서 1로 이루어진 가장 큰 정사각형을 찾아 넓이를 return 하는 solution 함수를 완성해 주세요. (단, 정사각형이란 축에 평행한 정사각형을 말합니다.)
예를 들어
1 | 2 | 3 | 4 |
---|---|---|---|
0 | 1 | 1 | 1 |
1 | 1 | 1 | 1 |
1 | 1 | 1 | 1 |
0 | 0 | 1 | 0 |
가 있다면 가장 큰 정사각형은
1 | 2 | 3 | 4 |
---|---|---|---|
0 | 1 | 1 | 1 |
1 | 1 | 1 | 1 |
1 | 1 | 1 | 1 |
0 | 0 | 1 | 0 |
가 되며 넓이는 9가 되므로 9를 반환해 주면 됩니다.
board | answer |
---|---|
[[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합니다.
def solution(board):
answer = 0
n = len(board)
m = len(board[0])
# dp 0행0열 값 setting
dp = [[0]*m for _ in range(n)]
dp[0] = board[0]
for i in range(1,n):
dp[i][0] = board[i][0]
for i in range(1,n):
for j in range(1,m):
if board[i][j] == 1:
dp[i][j] = min(dp[i-1][j-1],dp[i-1][j],dp[i][j-1]) + 1
for i in range(n):
tmp = max(dp[i])
answer = max(answer,tmp)
return answer*answer
랄것도 없고.. dp인지 캐치도 못했어서 이건 참고해서 푼 문제였다..
해당 블로그에서 많은 도움을 받았다.