알고리즘 공부/백준

백준 2810번: 컵홀더 (파이썬)

뚜써 2022. 1. 20. 03:20

https://www.acmicpc.net/problem/2810

 

2810번: 컵홀더

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

www.acmicpc.net

 

문제

 

십년이면 강산이 변한다.

 

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

 

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

 

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

 

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

 

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

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

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

 

입력

 

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

 

출력

 

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

 

풀이

 

가장 왼쪽 좌석부터 가장 오른쪽 좌석까지 각 좌석에서 컵홀더를 사용할 수 있는지 확인하면 된다.

 

dir = True일 때는 사람들이 모두 자신의 왼쪽 컵홀더를 사용하고, dir = False일 때는 사람들이 모두 자신의 오른쪽 컵 홀더를 사용한다고 하자. dir은 언제 바뀔까?

처음으로 커플석이 등장했을 때, 두 명이 모두 컵홀더를 사용하기 위해서는 왼쪽, 오른쪽 모두 사용하게 된다. 이때 dir이 바뀌게 된다. 한 번 변경된 dir은 바뀔 수 없다.

 

소스 코드

 

n = int(input())
seats = input()

cnt = 0
i = 0
dir = True
while i < n:
    if seats[i] == 'S':
        cnt += 1
        i += 1
    else:
        if dir:
            cnt += 2
            dir = not dir
        else:
            cnt += 1
        i += 2
print(cnt)