문제 출처 :
문제
세계는 균형이 잘 잡혀있어야 한다. 양과 음, 빛과 어둠 그리고 왼쪽 괄호와 오른쪽 괄호처럼 말이다.
정민이의 임무는 어떤 문자열이 주어졌을 때, 괄호들의 균형이 잘 맞춰져 있는지 판단하는 프로그램을 짜는 것이다. 문자열에 포함되는 괄호는 소괄호("()") 와 대괄호("[]")로 2종류이고, 문자열이 균형을 이루는 조건은 아래와 같다.
•
모든 왼쪽 소괄호("(")는 오른쪽 소괄호(")")와만 짝을 이뤄야 한다.
•
모든 왼쪽 대괄호("[")는 오른쪽 대괄호("]")와만 짝을 이뤄야 한다.
•
모든 오른쪽 괄호들은 자신과 짝을 이룰 수 있는 왼쪽 괄호가 존재한다.
•
모든 괄호들의 짝은 1:1 매칭만 가능하다. 즉, 괄호 하나가 둘 이상의 괄호와 짝지어지지 않는다.
•
짝을 이루는 두 괄호가 있을 때, 그 사이에 있는 문자열도 균형이 잡혀야 한다.
정민이를 도와 문자열이 주어졌을 때 균형잡힌 문자열인지 아닌지를 판단해 보자.
입력
각 문자열은 마지막 글자를 제외하고 영문 알파벳, 공백, 소괄호("( )"), 대괄호("[ ]")로 이루어져 있으며, 온점(".")으로 끝나고, 길이는 100글자보다 작거나 같다.
입력의 종료조건으로 맨 마지막에 온점 하나(".")가 들어온다.
출력
각 줄마다 해당 문자열이 균형을 이루고 있으면 "yes"를, 아니면 "no"를 출력한다.
예제 입력1
예제 출력1
So when I die (the [first] I will see in (heaven) is a score list).
[ first in ] ( first out ).
Half Moon tonight (At least it is better than no Moon at all].
A rope may form )( a trail in a maze.
Help( I[m being held prisoner in a fortune cookie factory)].
([ (([( [ ] ) ( ) (( ))] )) ]).
.
.
Plain Text
복사
yes
yes
no
no
no
yes
yes
Plain Text
복사
힌트
7번째의 " ."와 같이 괄호가 하나도 없는 경우도 균형잡힌 문자열로 간주할 수 있다.
문제 풀이
이 문제는 스택을 이용해서 해결할 수 있습니다.
자료구조에서 스택은 중요한 개념으로 스택의 개념을 이해하고 이를 활용하여 문제를 해결하는 방법을 배우는데 도움이 될 수 있는 문제입니다. 스택은 한 쪽 끝에서만 자료를 넣거나 뺄 수 있는 선형구조로 LIFO(Last In First Out)로 되어 있습니다. 자료를 넣는 것을 Push, 자료를 꺼내는 것을 Pop이라고 합니다.
출처 : 위키백과
1. 예제 입력: ( [] () )
1
2
3
4
5
6
7
8
9
2. 예제 입력: ( ( [ ) ] )
1
2
3
3. 입력 예제: ( [ ] ( )
4
1
2
3
4
5
6
7
소스 코드
while True: # 무한 루프를 시작합니다.
s = input() # 사용자로부터 입력을 받습니다.
if s == '.': # 입력이 '.'이면 루프를 종료합니다.
break
stack = [] # 스택을 초기화합니다.
temp = True # 괄호의 짝이 맞는지 확인하는 변수를 초기화합니다.
for i in s: # 입력받은 문자열의 각 문자에 대해 반복합니다.
if i == '(' or i == '[': # 여는 괄호이면 스택에 추가합니다.
stack.append(i)
elif i == ')': # 닫는 괄호 ')'이면
if not stack or stack[-1] == '[': # 스택이 비어있거나 스택의 마지막 요소가 '['이면 괄호의 짝이 맞지 않습니다.
temp = False
break
elif stack[-1] == '(': # 스택의 마지막 요소가 '('이면 짝이 맞으므로 스택에서 제거합니다.
stack.pop()
elif i == ']': # 닫는 괄호 ']'이면
if not stack or stack[-1] == '(': # 스택이 비어있거나 스택의 마지막 요소가 '('이면 괄호의 짝이 맞지 않습니다.
temp = False
break
elif stack[-1] == '[': # 스택의 마지막 요소가 '['이면 짝이 맞으므로 스택에서 제거합니다.
stack.pop()
if temp == True and not stack: # 모든 괄호의 짝이 맞고 스택이 비어있으면 'yes'를 출력합니다.
print('yes')
else: # 그렇지 않으면 'no'를 출력합니다.
print('no')
Python
복사