[Python] 백준 - 6549 히스토그램에서 가장 큰 직사각형

문제

 

히스토그램은 직사각형 여러 개가 아래쪽으로 정렬되어 있는 도형이다. 각 직사각형은 같은 너비를 가지고 있지만, 높이는 서로 다를 수도 있다. 예를 들어, 왼쪽 그림은 높이가 2, 1, 4, 5, 1, 3, 3이고 너비가 1인 직사각형으로 이루어진 히스토그램이다.

히스토그램에서 가장 넓이가 큰 직사각형을 구하는 프로그램을 작성하시오.

입력

입력은 테스트 케이스 여러 개로 이루어져 있다. 각 테스트 케이스는 한 줄로 이루어져 있고, 직사각형의 수 n이 가장 처음으로 주어진다. (1 ≤ n ≤ 100,000) 그 다음 n개의 정수 h1, ..., hn (0 ≤ hi ≤ 1,000,000,000)가 주어진다. 이 숫자들은 히스토그램에 있는 직사각형의 높이이며, 왼쪽부터 오른쪽까지 순서대로 주어진다. 모든 직사각형의 너비는 1이고, 입력의 마지막 줄에는 0이 하나 주어진다.

출력

각 테스트 케이스에 대해서, 히스토그램에서 가장 넓이가 큰 직사각형의 넓이를 출력한다.

 

문제해결방법 - stack, 분할정복 (서칭 참고)

 

1. stack을 이용하는 경우, 직사각형 높이 리스트 양 옆에 [0]을 더해줘서 누락되는 계산이 없도록 하는 점이 흥미롭다.

(활용가능성을 염두에 두자.) stack에는 인덱스를 저장하고 빼게 되며, 작아질 때마다 스택에 두었던 직사각형 높이 리스트를 길이와 더불어 곱하여 최댓값을 계속 갱신하는 구조로 진행하게 된다. 

 

2. 분할정복 재귀함수의 경우, mid, mid+1로 나누어 양 옆으로 인덱스를 진행하며, 직사각형의 높이 최소값에 길이를 곱해 최댓값을 갱신하는 방식으로 이루어진다. 갑자기 커지는 경우도 있으므로, 재귀함수로 같은 방법으로 진행하며 누락되는 계산이 없도록 한다.

 

- stack

import sys
input = sys.stdin.readline

while True:
	d = list(map(int, input().split()))
	if d[0] == 0:
		break
	d = [0]+d[1:]+[0]
	stack = [0]
	ans = 0
	for i in range(1, len(d)):
		while stack and d[i] < d[stack[-1]]:
			pop = stack.pop()
			lth = i-stack[-1]-1
			ans = max(ans, d[pop]*lth)
		stack.append(i)
	print(ans)

 

- 분할정복

import sys
sys.setrecursionlimit(10**6)
input = sys.stdin.readline

def sol(left, right):
	global d
	if left >= right:
		return d[left]
	mid = (left+right)//2
	lth = 2
	l, r = mid, mid+1
	m = min(d[l], d[r])
	area = m * lth
	A = sol(left, mid)
	B = sol(mid+1, right)
	while True:
		if (d[l] == 0 or left == l) and (d[r] == 0 or right == r):
			break
		if d[l] == 0 or left == l:
			if d[r+1] < m:
				m = d[r+1]
			r += 1
		elif d[r] == 0 or right == r:
			if d[l-1] < m:
				m = d[l-1]
			l -= 1
		else:
			if d[l-1] < d[r+1]:
				if d[r+1] < m:
					m = d[r+1]
				r += 1
			else:
				if d[l-1] < m:
					m = d[l-1]
				l -= 1
		lth += 1
		area = max(area, m*lth)
	return max(A, B, area)

while True:
	d = list(map(int, input().split()))
	n, d = d[0], d[1:]
	if n == 0:
		break
	print(sol(0, len(d)-1))