[한국정보올림피아드]쇠막대기

문제

문제풀이

---

• 여는 괄호일 경우 스택에 계속 저장
• 닫는 괄호일 경우에는 앞에 인덱스 요소에 따라 다른 연산 시행
• 앞에 인덱스 요소가 여는 괄호일 경우(레이저)는 레이저 한쌍에 해당하는 여는 괄호를 스택에서 삭제
• (삭제 된 스택의 크기) === (절단 된 쇠막대기 조각)
• 앞에 인덱스 요소가 닫는 괄호일 경우(쇠막대기 끝단)는 쇠막대기 중 가장 작은 막대기의 시작단을 스택에서 삭제
• 삭제 된 막대기 끝단에 조각 하나가 남아 있음

코드

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

  for (let i = 0; i < p.length; i++) {
    if (p[i] === ")") {
      if (p[i - 1] === "(") { // 레이저일 경우
        stack.pop(p[i - 1]);
        answer += stack.length;
      } else { //쇠막대기일 경우
        stack.pop();
        answer += 1; //쇠막대기의 남은 조각(끝단)을 가산
      }
    } else stack.push(p[i]);
  }

  return answer;
}

let pos = "()(((()())(())()))(())";
console.log(solution(pos));

---

function solution(s){
                let answer=0;
                let stack=[];
                for(let i=0; i<s.length; i++){
                    if(s[i]==='(') stack.push('(');
                    else{
                        stack.pop(); 
                        if(s[i-1]==='(') answer+=stack.length;
                        else answer++;
                        //stack.pop(); 이 위치에 하면 레이저까지 카운팅한다.
                    }
                }                          
                return answer;
            }
            
            let a="()(((()())(())()))(())";
            console.log(solution(a));