올바른 괄호

문제 설명

괄호가 바르게 짝지어졌다는 것은 '(' 문자로 열렸으면 반드시 짝지어서 ')' 문자로 닫혀야 한다는 뜻입니다. 예를 들어

• "()()" 또는 "(())()" 는 올바른 괄호입니다.
• ")()(" 또는 "(()(" 는 올바르지 않은 괄호입니다.

'(' 또는 ')' 로만 이루어진 문자열 s가 주어졌을 때, 문자열 s가 올바른 괄호이면 true를 return 하고, 올바르지 않은 괄호이면 false를 return 하는 solution 함수를 완성해 주세요.

제한사항

• 문자열 s의 길이 : 100,000 이하의 자연수
• 문자열 s는 '(' 또는 ')' 로만 이루어져 있습니다.

핵심 아이디어

해당 문제는 stack을 활용해서 풀어야 한다. '('이 나오면 스택에 삽입하고 ')'이 나오면 스택에서 제거하는 식으로 짝을 맞춰 나간다. 스택은 후입선출의 구조를 갖고 있다. 즉, 먼저 들어간 요소가 가장 마지막에 나오는 형태로 동작한다. push()와 pop() 을 통해 삽입, 삭제 연산을 수행하면 모두 O(1)의 시간복잡도가 걸린다. 삽입은 스택에 맨 위에 저장하고 삭제 역시 스택의 맨 마지막 요소를 제거한다. 

코드 구현

def solution(s):
    
    # 스택을 초기화
    stack = []
    # 문자열 하나씩 순회하면서 탐색한다.
    for i in s:
        if i == '(':
            # 스택에 삽입한다.
            stack.append(i)
        elif i == ')':
            # 스택이 비어있지 않고 마지막 요소가 짝 이라면
            if len(stack) != 0 and stack[-1] == '(':
                stack.pop()
            else:
                # 스택에 넣고 반복문 종료한다.
                stack.append(i)
                break
    # 모든 문자열을 탐색 후 스택을 확인한다.
    if len(stack) == 0:
        answer = True
    else:
        answer = False
                

    return answer