[백준] 4949 균형잡힌 세상 Python

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

 

📝  문제

세계는 균형이 잘 잡혀있어야 한다. 양과 음, 빛과 어둠 그리고 왼쪽 괄호와 오른쪽 괄호처럼 말이다.
정민이의 임무는 어떤 문자열이 주어졌을 때, 괄호들의 균형이 잘 맞춰져 있는지 판단하는 프로그램을 짜는 것이다.
문자열에 포함되는 괄호는 소괄호("()") 와 대괄호("[]")로 2종류이고, 문자열이 균형을 이루는 조건은 아래와 같다.

- 모든 왼쪽 소괄호("(")는 오른쪽 소괄호(")")와만 짝을 이뤄야 한다.
- 모든 왼쪽 대괄호("[")는 오른쪽 대괄호("]")와만 짝을 이뤄야 한다.
- 모든 오른쪽 괄호들은 자신과 짝을 이룰 수 있는 왼쪽 괄호가 존재한다.
- 모든 괄호들의 짝은 1:1 매칭만 가능하다. 즉, 괄호 하나가 둘 이상의 괄호와 짝지어지지 않는다.
- 짝을 이루는 두 괄호가 있을 때, 그 사이에 있는 문자열도 균형이 잡혀야 한다.

정민이를 도와 문자열이 주어졌을 때 균형잡힌 문자열인지 아닌지를 판단해보자.

 

🔎  풀이

문자열을 처음부터 2개씩 슬라이싱 하여 "()"나 "[]"와 일치할 경우 해당 문자열을 제외한 나머지 문자열을 변수에 대입한다.

(문자열 갯수 // 2)번 실행했을 때, 변수에 값이 남아있으면 "no"를, 없으면 "yes"를 출력한다.

 

💻  코드

while True:
    text = input()
    compare = ""

    if text == ".":
        break

    for i in text:
        if i == "(" or i == ")" or i ==  "[" or i == "]":
            compare += i
    
    for j in range(len(compare) // 2):     
        for jj in range(len(compare)):
            if compare[jj:jj+2] == "()" or compare[jj:jj+2] == "[]":
                compare = compare[0:jj] + compare[jj+2:]
                break
    
    print("no" if compare else "yes")