파이썬 그리디 알고리즘 백준 2810 컵홀더

문제

십년이면 강산이 변한다.

강산이네 동네에 드디어 극장이 생겼고, 강산이는 극장에 놀러갔다. 매점에서 콜라를 산 뒤, 자리에 앉은 강산이는 큰 혼란에 빠졌다. 양쪽 컵홀더를 이미 옆 사람들이 차지했기 때문에 콜라를 꽂을 컵 홀더가 없었기 때문이다. 영화를 보는 내내 콜라를 손에 들고 있던 강산이는 극장에 다시 왔을 때는 꼭 콜라를 컵 홀더에 놓겠다는 다짐을 한 후 집에 돌아갔다.

극장의 한 줄에는 자리가 N개가 있다. 서로 인접한 좌석 사이에는 컵홀더가 하나씩 있고, 양 끝 좌석에는 컵홀더가 하나씩 더 있다. 또, 이 극장에는 커플석이 있다. 커플석 사이에는 컵홀더가 없다.

극장의 한 줄의 정보가 주어진다. 이때, 이 줄에 사람들이 모두 앉았을 때, 컵홀더에 컵을 꽂을 수 있는 최대 사람의 수를 구하는 프로그램을 작성하시오. 모든 사람은 컵을 한 개만 들고 있고, 자신의 좌석의 양 옆에 있는 컵홀더에만 컵을 꽂을 수 있다.

S는 일반 좌석, L은 커플석을 의미하며, L은 항상 두개씩 쌍으로 주어진다.

어떤 좌석의 배치가 SLLLLSSLL일때, 컵홀더를 *로 표시하면 아래와 같다.

*S*LL*LL*S*S*LL*

위의 예에서 적어도 두 명은 컵홀더를 사용할 수 없다.

입력

첫째 줄에 좌석의 수 N이 주어진다. (1 ≤ N ≤ 50) 둘째 줄에는 좌석의 정보가 주어진다.

출력

컵을 컵홀더에 놓을 수 있는 최대 사람의 수를 출력한다.

풀이

컵홀더를 놓을 수 있는 최대 사람의 수를 구하기 위해 전체 컵홀더 수를 세고

사람 수 보다 컵홀더 수가 많으면 사람 수를 출력하고, 사람 수 가 더 많으면 컵 홀더 수를 출력하면 된다.

전체 컵홀더 수를 구하기 위해서 입력받은 S,와 LL 정보를 정렬한다.

가장 왼쪽의 컵홀더를 계산하여 ans를 1로 초기화 하고 좌석별로 왼쪽의 컵홀더 수를 더하면 된다.

그 후 heapq에서 하나씩 꺼내서 S일 경우 1개를 더하고

L일 경우 L은 항상 2개씩 세트이기 때문에 1개를 더하고 heapq.pop을 한번 더 수행해준다.

ans가 사람 수 와 같다면 반복을 멈추고 그대로 출력한다.

heapq.pop으로 모든 원소가 없어지면 종료된다.

 

import sys
import heapq
n = int(sys.stdin.readline())
seatStrings = list(sys.stdin.readline().rstrip("\n"))
seatStrings = sorted(seatStrings,reverse=True)
heapq.heapify(seatStrings)
ans=1
while seatStrings:
    temp = heapq.heappop(seatStrings)
    if(temp=="L"):
        ans+=1
        heapq.heappop(seatStrings)
    elif(temp=="S"):
        ans+=1
    if(ans==n):
        break

print(ans)