codility - Brackets 정답 및 해설

난이도 ★★☆☆☆

문제를 보시려면 여기를 클릭

문제 요약

괄호의 짝이 정상적으로 맞는지를 리턴. 단, 빈 문자열 입력시에도 정상이라고 리턴할 것

힌트

Stack 을 이용하는 전형적인 문제  1. 여는 괄호 (, {, [ 는 push  2. 닫는 괄호 ), }, ] 는 pop  3. stack 이 비어있을 때 pop 하면 에러 발생함. 주의할 것

이하는 코드입니다.

public int solution(String S) {
    Stack<Character> stack = new Stack<>();
    
    for (int i = 0; i < S.length(); i++) {
        char c = S.charAt(i);
        
        if (c == '(' || c == '{' || c == '[') {
            stack.push(c);
        }
        else {
            if (stack.isEmpty()) {
                return 0;
            }
            
            char lastC = stack.pop();
            
            if (c == ')' && lastC != '(') {
                return 0;
            }
            
            if (c == '}' && lastC != '{') {
                return 0;
            }
            
            if (c == ']' && lastC != '[') {
                return 0;
            }
        }
    }
    
    if (!stack.isEmpty()) {
        return 0;
    }
    
    return 1;
}