[JavaScript/section 6] 05 - 쇠막대기

📌 05 - 쇠막대기(스택)

쇠막대기와 레이저의 배치를 나타내는 괄호 표현이 주어졌을 때, 잘려진 쇠막대기 조각의 총 개수를 구하는 문제이다.

쇠 막대기가 레이저를 만날때마다 몇 개의 조각이 발생한다. 이러한 조각이 생길때마다 바구니에 넣어준다라고 생각하면 문제를 이해하는데 많은 도움이 된다.

풀이 방법

• for 문을 돌면서 '('를 만나면 무조건 스택에 넣는다.
• ')'를 만나면 pop() 함수를 이용해 스택에서 자료를 꺼내는데, 여기서 s[i - 1]의 값이 '('이라면 여는 괄호와 닫는 괄호의 인접한 쌍으로 레이저에 해당한다. 스택의 길이가 쇠막대기의 개수이므로 스택의 길이를 answer 변수(바구니)에 누적시킨다.
• s[i - 1]의 값이 ')'이라면 쇠막대기의 오른쪽 끝을 나타내므로 answer++을 해준다. 쇠막대기의 마지막 조각을 바구니에 넣어주는 것과 같다.

 

📝 풀이

function solution(s) {
  let answer = 0;
  let stack = [];

  for (let i = 0; i < s.length; i++) {
    if (s[i] === '(') stack.push(s[i]);
    else {
      stack.pop();
      if (s[i - 1] === '(') {
        answer += stack.length;
      } else {
        answer++;
      }
    }
  }

  return answer;
}

let a = '(((()(()()))(())()))(()())';
console.log(solution(a));